Searched refs:mutable_dims (Results 1 – 20 of 20) sorted by relevance
93 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize()241 *(output_array.mutable_shape()->mutable_dims()) = specified_output_shape; in ProcessTransposeConvOperator()365 *(output_array.mutable_shape()->mutable_dims()) = dims; in ProcessOpWithShapeInput()458 *output_shape.mutable_dims() = shape_data; in ProcessTensorFlowReshapeOperator()589 auto* mutable_dims = output_array.mutable_shape()->mutable_dims(); in ProcessTensorFlowReductionOperator() local590 mutable_dims->clear(); in ProcessTensorFlowReductionOperator()594 mutable_dims->emplace_back(1); in ProcessTensorFlowReductionOperator()597 mutable_dims->emplace_back(input_shape.dims(i)); in ProcessTensorFlowReductionOperator()637 *output_array.mutable_shape()->mutable_dims() = output_dims; in ProcessSliceOperator()690 auto& output_dims = *output_shape.mutable_dims(); in ProcessConcatenationOperator()[all …]
49 *(reshape_array.mutable_shape()->mutable_dims()) = { in CreateReshapeFromReorderAxes()78 *(perm_array.mutable_shape()->mutable_dims()) = { in CreateTransposeFromReorderAxes()
68 padded_shape.mutable_dims()->insert(padded_shape.mutable_dims()->begin(), in Slice()
223 *(transpose_array.mutable_shape()->mutable_dims()) = { in Run()236 *(reshape_array.mutable_shape()->mutable_dims()) = { in Run()
65 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
58 bias_array.mutable_shape()->mutable_dims()->push_back(depth); in ProcessLinearOperator()
76 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
96 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
61 output_shape.mutable_dims()->at(axis) = 1; in ReduceGeneric()91 output_shape.mutable_dims()->swap(new_dims); in ReduceGeneric()
116 *weights_array.mutable_shape()->mutable_dims() = {1, width, height, depth}; in Run()
97 *mean_shape.mutable_dims() = multiplier_shape.dims(); in Run()
194 *transpose_array.mutable_shape()->mutable_dims() = in TransposeInput()
38 *input0.mutable_shape()->mutable_dims() = input_shape; in RunResolveSum()42 *input1.mutable_shape()->mutable_dims() = {static_cast<int>(axis.size())}; in RunResolveSum()46 *output.mutable_shape()->mutable_dims() = output_shape; in RunResolveSum()
129 std::vector<int>* in_array_shape_dim = in_array_shape->mutable_dims(); in PrepareModel()148 std::vector<int>* out_array_shape_dim = out_array_shape->mutable_dims(); in PrepareModel()
50 *(array_shape->mutable_dims()) = shape; in CreateArray()
71 array.mutable_shape()->mutable_dims()->clear(); in ImportTensors()74 array.mutable_shape()->mutable_dims()->push_back(d); in ImportTensors()
113 weight_array_shape->mutable_dims(); in BuildQuantizableTestModel()
613 auto* shape_dims = shape->mutable_dims(); in ExtendShape()624 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape()1492 *shape->mutable_dims() = dims; in CreateOrCheckRnnStateArray()1641 auto& input_array_dims = *input_array.mutable_shape()->mutable_dims(); in ResolveModelFlags()1652 *input_array.mutable_shape()->mutable_dims(); in ResolveModelFlags()2063 std::vector<int>* output_dims = output_shape->mutable_dims(); in ShuffleDims()2337 array.mutable_shape()->mutable_dims()->push_back(dim); in UseArraysExtraInfo()
226 shape->mutable_dims()->clear(); in ImportShape()231 *shape->mutable_dims() = input_dims_only_sizes; in ImportShape()536 array.mutable_shape()->mutable_dims()->emplace_back(data.size()); in CreateConstArray()1489 auto& dst_array_dims = *array.mutable_shape()->mutable_dims(); in ConvertPlaceholderOperator()
334 std::vector<int>* mutable_dims() { return &dims_; }