Home
last modified time | relevance | path

Searched refs:parameters_size (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc77 arg_shapes.resize(program_shape.parameters_size()); in GetShardedProgramShapes()
85 if (instr.parameter_number() >= program_shape.parameters_size()) { in GetShardedProgramShapes()
88 instr.parameter_number(), program_shape.parameters_size()); in GetShardedProgramShapes()
160 } else if (argument_layouts->size() != program_shape.parameters_size()) { in DetermineArgumentLayoutsFromCompileOptions()
164 argument_layouts->size(), program_shape.parameters_size()); in DetermineArgumentLayoutsFromCompileOptions()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc81 layouts.reserve(program_shape->parameters_size()); in RealMain()
82 for (int i = 0; i < program_shape->parameters_size(); ++i) { in RealMain()
Ddumped_computation_to_text.cc57 layouts.reserve(program_shape->parameters_size()); in RealMain()
58 for (int i = 0; i < program_shape->parameters_size(); ++i) { in RealMain()
Dreplay_computation.cc119 computation.proto().host_program_shape().parameters_size()); in CompileExecutable()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_test.cc164 ASSERT_EQ(program_shape.parameters_size(), in TEST_F()
165 program_shape_copy.parameters_size()); in TEST_F()
166 for (int i = 0; i < program_shape.parameters_size(); ++i) { in TEST_F()
Dshape.cc219 std::vector<string> parameter_strings(parameters_size()); in ToString()
220 for (int i = 0; i < parameters_size(); ++i) { in ToString()
Dshape.h290 int parameters_size() const { return parameters_.size(); } in parameters_size() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_util.cc53 if (program_shape.parameters_size() != argument_shapes_size) { in CreateModuleConfig()
55 program_shape.parameters_size(), in CreateModuleConfig()
Dlocal_service.cc110 if (argument_layouts.size() != program_shape.parameters_size()) { in CompileExecutables()
113 program_shape.parameters_size(), argument_layouts.size()); in CompileExecutables()
Dshape_inference.cc66 if (reducer_shape.parameters_size() != inputs * 2) { in VerifyReducerShape()
70 inputs * 2, reducer_shape.parameters_size()); in VerifyReducerShape()
1171 if (arg_shapes.size() != to_apply.parameters_size()) { in InferMapShape()
1175 to_apply.parameters_size(), arg_shapes.size()); in InferMapShape()
1186 for (int i = 0; i < to_apply.parameters_size(); ++i) { in InferMapShape()
2251 if (select_shape.parameters_size() != 2) { in InferSelectAndScatterShape()
2255 select_shape.parameters_size()); in InferSelectAndScatterShape()
2750 if (condition.parameters_size() != 1) { in InferWhileShape()
2752 condition.parameters_size()); in InferWhileShape()
2754 if (body.parameters_size() != 1) { in InferWhileShape()
[all …]
Dhlo_module.cc341 TF_RET_CHECK(expected_program_shape.parameters_size() == in CreateFromProto()
343 for (int i = 0; i < expected_program_shape.parameters_size(); ++i) { in CreateFromProto()
Dservice.cc1058 if (arg->computation().host_program_shape().parameters_size() != 0) { in ComputeConstantGraph()
/external/tensorflow/tensorflow/compiler/aot/
Dcompile.cc79 std::vector<const xla::Shape*> arg_layout_ptrs(pshape->parameters_size()); in CompileXla()
80 std::vector<xla::Shape> arg_layouts(pshape->parameters_size()); in CompileXla()
81 for (int i = 0; i < pshape->parameters_size(); ++i) { in CompileXla()
Dcodegen.cc175 const int num_args = ps.parameters_size(); in GenArgMethods()
277 const int num_args = ps.parameters_size(); in GenVariableMethods()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc153 config.program_shape().parameters_size()); in Compile()
155 config.program_shape().parameters_size()); in Compile()
156 for (int i = 0; i < config.program_shape().parameters_size(); ++i) { in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.cc112 arg_shapes.reserve(program_shape->parameters_size()); in Compile()
113 for (int i = 0; i < program_shape->parameters_size(); ++i) { in Compile()
Dxla_jit_compiled_cpu_function_test.cc221 ASSERT_EQ(program_shape.parameters_size(), 2); in TEST()
272 ASSERT_EQ(program_shape.parameters_size(), 2); in TEST()
Dxla_compiler_test.cc1292 ASSERT_EQ(program_shape->parameters_size(), 2); in TEST_F()
1363 ASSERT_EQ(program_shape->parameters_size(), 2); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc64 if (lhs.parameters_size() != rhs.parameters_size()) { in ProgramShapesEqual()
67 for (int i = 0; i < lhs.parameters_size(); i++) { in ProgramShapesEqual()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc67 if (program_shape.parameters_size() != argument_shapes.size()) { in CreateModuleConfig()
69 program_shape.parameters_size(), in CreateModuleConfig()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_optimization_parameters_utils.cc157 const int num_inputs = program_shape.parameters_size(); in GetBaseAuxiliaryParameterCount()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc370 for (int64 i = 0; i < input_program_shape.parameters_size(); ++i) { in XlaCompiledProgramShape()
1120 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
1498 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
1545 EXPECT_EQ(program_shape.parameters_size(), 1); in TEST()
2035 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc642 ASSERT_EQ(muladd_shape->parameters_size(), 2); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3691 TF_RET_CHECK(program_shape.parameters_size() == in Run()
3692 new_program_shape.parameters_size()) in Run()
3694 for (int64 i = 0; i < program_shape.parameters_size(); ++i) { in Run()
3702 for (int64 i = 0; i < new_program_shape.parameters_size(); ++i) { in Run()
/external/tensorflow/tensorflow/core/framework/
Dmodel.cc1533 int64 num_parameters = node_proto.parameters_size(); in FromProtoHelper()

12