Searched defs:intermediate_shape (Results 1 – 3 of 3) sorted by relevance
67 auto intermediate_shape = output_shape; in CompileGrad() local
167 Shape intermediate_shape = intermediate_array.shape(); in Run() local
1284 std::vector<int32_t> intermediate_shape(input_tensor.dims->size - 1); in TransformUnpackIntoSupportedOps() local