| /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 | 43 *input0.mutable_shape()->mutable_dims() = input_shape; in RunResolveSum()
47 *input1.mutable_shape()->mutable_dims() = {static_cast<int>(axis.size())}; in RunResolveSum()
51 *output.mutable_shape()->mutable_dims() = output_shape; in RunResolveSum()
|
| D | unpack_quantize_test.cc | 61 Shape* in_array_shape = in_array.mutable_shape(); in PrepareModel()
87 Shape* out_array_shape = out_array.mutable_shape(); in PrepareModel()
|
| D | resolve_constant_concatenation_test.cc | 131 Shape* in_array_shape = in_array.mutable_shape(); in PrepareModel()
150 Shape* out_array_shape = out_array.mutable_shape(); in PrepareModel()
|
| /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()
240 *(output_array.mutable_shape()->mutable_dims()) = specified_output_shape; in ProcessTransposeConvOperator()
290 model->GetArray(output_name).mutable_shape(), in ProcessDepthwiseConvOperator()
364 *(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()
|
| /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/compiler/xla/client/ |
| D | xla_builder.cc | 106 *instr.mutable_shape() = shape.ToProto(); in BuildAddDependency()
122 *instr.mutable_shape() = program_shape.result().ToProto(); in BuildFusion()
133 *instr.mutable_shape() = shape.ToProto(); in BuildBitcast()
146 *instr.mutable_shape() = shape.ToProto(); in BuildDomain()
156 *instr.mutable_shape() = shape.ToProto(); in BuildPartitionId()
168 *instr.mutable_shape() = shape.ToProto(); in BuildRngGetAndUpdateState()
457 *instr.mutable_shape() = param_shape.ToProto(); in SetDynamicBinding()
539 *instructions_[index].mutable_shape() = in Build()
619 *instr.mutable_shape() = shape.ToProto(); in InDimBroadcast()
772 *instr.mutable_shape() = shape.ToProto(); in BinaryOpNoBroadcast()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_conversion_folding.cc | 77 hlo->mutable_shape()->set_element_type(BF16); in FoldOutputConversions()
78 bfloat16_conversion_folding_->UpdateLayout(hlo->mutable_shape()); in FoldOutputConversions()
243 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}) in HandleAllReduce()
246 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i})); in HandleAllReduce()
|
| D | collective_decomposer_utils.cc | 42 update_layout(*zero->mutable_shape()); in CreateStartIndicesForCollectiveDecomposition()
140 update_layout(*table->mutable_shape()); in CreateStartIndicesForCollectiveDecomposition()
146 update_layout(*ds->mutable_shape()); in CreateStartIndicesForCollectiveDecomposition()
|
| D | bfloat16_normalization.cc | 177 hlo->mutable_shape(), [&](Shape* subshape, const xla::ShapeIndex& index) { in ChangeOutputTypeThenInsertConvertBack()
182 bfloat16_normalization_->UpdateLayout(hlo->mutable_shape()); in ChangeOutputTypeThenInsertConvertBack()
350 auto subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), {i}); in HandleMultipleOutputs()
370 *tuple->mutable_shape() = hlo->shape(); in HandleMultipleOutputs()
378 *tuple->mutable_shape() = original_shape; in HandleMultipleOutputs()
|
| D | layout_assignment.cc | 1025 LayoutUtil::ClearLayout(tuple_copy->mutable_shape()); in CreateCopyWithNewLayout()
1027 shape_with_layout, tuple_copy->mutable_shape())); in CreateCopyWithNewLayout()
1035 LayoutUtil::ClearLayout(copy->mutable_shape()); in CreateCopyWithNewLayout()
1037 shape_with_layout, copy->mutable_shape())); in CreateCopyWithNewLayout()
1073 *param->mutable_shape() = operand->shape(); in CopyOperandIfLayoutsDiffer()
1921 fusion_operand->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()
1928 fusion->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()
1935 fused_instruction->mutable_shape())); in SetFusionLayouts()
1940 fused_instruction->mutable_shape())); in SetFusionLayouts()
1946 LayoutUtil::ClearLayout(fused_instruction->mutable_shape()); in SetFusionLayouts()
[all …]
|
| D | select_and_scatter_expander_test.cc | 54 inst->mutable_shape()->clear_layout(); in ClearInstructionLayout()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | attr_util.cc | 179 auto* mutable_shape = mutable_list->mutable_shape(); in FillAttrValueMapUsingAggregate() local
180 mutable_shape->Reserve(num_values); in FillAttrValueMapUsingAggregate()
228 auto* mutable_shape = attr_tmp.mutable_shape(); in FillAttrValueMapUsingScalar() local
232 mutable_shape->mutable_dim()->Reserve(rank); in FillAttrValueMapUsingScalar()
234 mutable_shape->add_dim()->set_size(shape[d]); in FillAttrValueMapUsingScalar()
237 mutable_shape->set_unknown_rank(true); in FillAttrValueMapUsingScalar()
551 tf_shape.AsProto(tf_attr->mutable_shape()); in SetUpScalarAttr()
554 unranked_shape.AsProto(tf_attr->mutable_shape()); in SetUpScalarAttr()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 70 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
87 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
104 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 1539 p.mutable_shape()->set_element_type(PRED); in TEST_F()
1541 p.mutable_shape()->clear_dimensions(); in TEST_F()
1542 p.mutable_shape()->add_dimensions(len); in TEST_F()
1543 SetDefaultLayoutOnProto(p.mutable_shape()); in TEST_F()
1589 p.mutable_shape()->set_element_type(F16); in TEST_F()
1590 p.mutable_shape()->clear_dimensions(); in TEST_F()
1591 p.mutable_shape()->add_dimensions(4); in TEST_F()
1592 SetDefaultLayoutOnProto(p.mutable_shape()); in TEST_F()
1611 p.mutable_shape()->set_element_type(U16); in TEST_F()
1612 p.mutable_shape()->clear_dimensions(); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_hlo_test.cc | 83 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P()
88 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/compiler/xla/service/gpu/ |
| D | reduction_layout_normalizer.cc | 59 LayoutUtil::ClearLayout(copy->mutable_shape()); in HandleReduce()
61 first_instruction_shape, copy->mutable_shape())); in HandleReduce()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_attributes.cc | 98 SetTensorShapeProto(attr, value->mutable_shape()); in ConvertAttribute()
283 SetTensorShapeProto(shaped_type, value.mutable_shape()); in SetShapeAttribute()
489 handle_data->mutable_shape()); in ConvertHandleData()
491 handle_data->mutable_shape()->set_unknown_rank(true); in ConvertHandleData()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_handle_test.cc | 78 dtypes_and_shapes->mutable_shape()->MergeFrom(shape); in TEST_F()
|