| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_normalization_test.cc | 36 *input0.mutable_shape()->mutable_dims() = input_shape; in RunIdentifyL2Normalization()
40 *output.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
47 *sumoutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
55 *sqrtoutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
62 *divoutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
75 *rsqoutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
82 *muloutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Normalization()
|
| D | identify_l2_pool_test.cc | 35 *input0.mutable_shape()->mutable_dims() = input_shape; in RunIdentifyL2Pool()
39 *output.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Pool()
46 *avgpooloutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Pool()
53 *sqrtoutput.mutable_shape()->mutable_dims() = output_shape; in RunIdentifyL2Pool()
|
| D | resolve_constant_unary_test.cc | 40 *input0.mutable_shape()->mutable_dims() = input_shape; in RunResolveSum()
44 *input1.mutable_shape()->mutable_dims() = {static_cast<int>(axis.size())}; in RunResolveSum()
48 *output.mutable_shape()->mutable_dims() = output_shape; in RunResolveSum()
|
| D | unpack_quantize_test.cc | 60 Shape* in_array_shape = in_array.mutable_shape(); in PrepareModel()
86 Shape* out_array_shape = out_array.mutable_shape(); in PrepareModel()
|
| D | resolve_constant_concatenation_test.cc | 130 Shape* in_array_shape = in_array.mutable_shape(); in PrepareModel()
149 Shape* out_array_shape = out_array.mutable_shape(); in PrepareModel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander_test.cc | 62 scatter_operand->mutable_shape()->clear_layout(); in TEST_F()
97 scatter_operand->mutable_shape()->clear_layout(); in TEST_F()
130 scatter_operand->mutable_shape()->clear_layout(); in TEST_F()
|
| D | bfloat16_conversion_folding.cc | 78 hlo->mutable_shape()->set_element_type(BF16); in FoldOutputConversions()
79 bfloat16_conversion_folding_->UpdateLayout(hlo->mutable_shape()); in FoldOutputConversions()
244 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}) in HandleAllReduce()
247 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i})); in HandleAllReduce()
|
| D | bfloat16_normalization.cc | 178 hlo->mutable_shape(), [&](Shape* subshape, const xla::ShapeIndex& index) { in ChangeOutputTypeThenInsertConvertBack()
183 bfloat16_normalization_->UpdateLayout(hlo->mutable_shape()); in ChangeOutputTypeThenInsertConvertBack()
351 auto subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), {i}); in HandleMultipleOutputs()
371 *tuple->mutable_shape() = hlo->shape(); in HandleMultipleOutputs()
379 *tuple->mutable_shape() = original_shape; in HandleMultipleOutputs()
|
| D | layout_assignment.cc | 835 LayoutUtil::ClearLayout(tuple_copy->mutable_shape()); in CreateCopyWithNewLayout()
837 shape_with_layout, tuple_copy->mutable_shape())); in CreateCopyWithNewLayout()
845 LayoutUtil::ClearLayout(copy->mutable_shape()); in CreateCopyWithNewLayout()
847 shape_with_layout, copy->mutable_shape())); in CreateCopyWithNewLayout()
882 *param->mutable_shape() = operand->shape(); in CopyOperandIfLayoutsDiffer()
1690 fusion_operand->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()
1697 fusion->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()
1704 fused_instruction->mutable_shape())); in SetFusionLayouts()
1709 fused_instruction->mutable_shape())); in SetFusionLayouts()
1715 LayoutUtil::ClearLayout(fused_instruction->mutable_shape()); in SetFusionLayouts()
[all …]
|
| D | bfloat16_propagation.cc | 801 UpdateLayout(new_constant->mutable_shape()); in ResolveConvertedConstants()
931 inst->mutable_shape(), leaf_index))) { in Run()
984 Shape* subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index); in OutputTypeAfterChange()
1007 entry.emplace(ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index), in AddToOrRemoveFromBF16ChangeSet()
1016 ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index)); in AddToOrRemoveFromBF16ChangeSet()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 94 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize()
151 output_array.mutable_shape(), in ProcessConvOperator()
242 *(output_array.mutable_shape()->mutable_dims()) = specified_output_shape; in ProcessTransposeConvOperator()
292 model->GetArray(output_name).mutable_shape(), in ProcessDepthwiseConvOperator()
366 *(output_array.mutable_shape()->mutable_dims()) = dims; in ProcessOpWithShapeInput()
458 auto& output_shape = *output_array.mutable_shape(); in ProcessTensorFlowReshapeOperator()
590 auto* mutable_dims = output_array.mutable_shape()->mutable_dims(); in ProcessTensorFlowReductionOperator()
638 *output_array.mutable_shape()->mutable_dims() = output_dims; in ProcessSliceOperator()
650 Shape* output_shape = model->GetArray(output_name).mutable_shape(); in ProcessReorderAxesOperator()
690 auto& output_shape = *output_array.mutable_shape(); in ProcessConcatenationOperator()
[all …]
|
| D | convert_reorder_axes.cc | 49 *(reshape_array.mutable_shape()->mutable_dims()) = { in CreateReshapeFromReorderAxes()
79 *(perm_array.mutable_shape()->mutable_dims()) = { in CreateTransposeFromReorderAxes()
|
| D | reorder_reshape_transpose.cc | 223 *(transpose_array.mutable_shape()->mutable_dims()) = { in Run()
236 *(reshape_array.mutable_shape()->mutable_dims()) = { in Run()
|
| D | resolve_constant_shape_or_rank.cc | 61 output_array.mutable_shape()->ReplaceDims( in Run()
|
| D | convert_trivial_pack_to_reshape.cc | 66 *(shape_array.mutable_shape()->mutable_dims()) = {shape_array_dims}; in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 94 *instr->mutable_shape() = shape.ToProto(); in SetInstructionAsConstant()
118 *new_param->mutable_shape() = ConvertShapeProtoToPred(inst.shape()); in CopyReducer()
128 *new_root->mutable_shape() = ConvertShapeProtoToPred(inst.shape()); in CopyReducer()
164 *instr.mutable_shape() = program_shape.result().ToProto(); in BuildFusion()
175 *instr.mutable_shape() = shape.ToProto(); in BuildBitcast()
394 *instr.mutable_shape() = param_shape.ToProto(); in SetDynamicBinding()
476 *instructions_[index].mutable_shape() = in Build()
556 *instr.mutable_shape() = shape.ToProto(); in InDimBroadcast()
706 *instr.mutable_shape() = shape.ToProto(); in BinaryOpNoBroadcast()
725 *instr.mutable_shape() = shape.ToProto(); in Compare()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 108 multi_array->mutable_shape()->Add(batch_size);
110 multi_array->mutable_shape()->Add(depth_size);
111 multi_array->mutable_shape()->Add(height_size);
112 multi_array->mutable_shape()->Add(width_size);
163 multi_array->mutable_shape()->Add(batch_size);
165 multi_array->mutable_shape()->Add(depth_size);
166 multi_array->mutable_shape()->Add(height_size);
167 multi_array->mutable_shape()->Add(width_size);
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 69 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
86 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
103 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 1531 p.mutable_shape()->set_element_type(PRED); in TEST_F()
1533 p.mutable_shape()->clear_dimensions(); in TEST_F()
1534 p.mutable_shape()->add_dimensions(len); in TEST_F()
1535 SetDefaultLayoutOnProto(p.mutable_shape()); in TEST_F()
1581 p.mutable_shape()->set_element_type(F16); in TEST_F()
1582 p.mutable_shape()->clear_dimensions(); in TEST_F()
1583 p.mutable_shape()->add_dimensions(4); in TEST_F()
1584 SetDefaultLayoutOnProto(p.mutable_shape()); in TEST_F()
1603 p.mutable_shape()->set_element_type(U16); in TEST_F()
1604 p.mutable_shape()->clear_dimensions(); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_hlo_test.cc | 84 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P()
89 reduce_instruction->mutable_operand(0)->mutable_shape(); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 59 tensor_prop.mutable_shape()->add_dim()->set_size(1); in GetTensorProperties()
66 tensor_prop.mutable_shape()->add_dim()->set_size(size); in GetTensorProperties()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test.cc | 203 variable->mutable_shape()->add_dim()->set_size(1); in TEST()
208 variable2->mutable_shape()->add_dim()->set_size(1); in TEST()
213 variable3->mutable_shape()->add_dim()->set_size(5); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transfer_utils.cc | 75 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo()
91 *tensor_shape_type_proto.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import.cc | 72 array.mutable_shape()->mutable_dims()->clear(); in ImportTensors()
75 array.mutable_shape()->mutable_dims()->push_back(d); in ImportTensors()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 112 auto* shape = value->mutable_shape(); in ConvertAttribute()
154 mangling_util::DemangleShape(attr_value, value->mutable_shape())); in ConvertAttribute()
423 *value.mutable_shape() = tshape; in SetShapeAttribute()
|