/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | utils.cc | 77 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/ |
D | dumped_computation_to_operation_list.cc | 81 layouts.reserve(program_shape->parameters_size()); in RealMain() 82 for (int i = 0; i < program_shape->parameters_size(); ++i) { in RealMain()
|
D | dumped_computation_to_text.cc | 57 layouts.reserve(program_shape->parameters_size()); in RealMain() 58 for (int i = 0; i < program_shape->parameters_size(); ++i) { in RealMain()
|
D | replay_computation.cc | 119 computation.proto().host_program_shape().parameters_size()); in CompileExecutable()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | shape_test.cc | 164 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()
|
D | shape.cc | 219 std::vector<string> parameter_strings(parameters_size()); in ToString() 220 for (int i = 0; i < parameters_size(); ++i) { in ToString()
|
D | shape.h | 290 int parameters_size() const { return parameters_.size(); } in parameters_size() function
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module_util.cc | 53 if (program_shape.parameters_size() != argument_shapes_size) { in CreateModuleConfig() 55 program_shape.parameters_size(), in CreateModuleConfig()
|
D | local_service.cc | 110 if (argument_layouts.size() != program_shape.parameters_size()) { in CompileExecutables() 113 program_shape.parameters_size(), argument_layouts.size()); in CompileExecutables()
|
D | shape_inference.cc | 66 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 …]
|
D | hlo_module.cc | 341 TF_RET_CHECK(expected_program_shape.parameters_size() == in CreateFromProto() 343 for (int i = 0; i < expected_program_shape.parameters_size(); ++i) { in CreateFromProto()
|
D | service.cc | 1058 if (arg->computation().host_program_shape().parameters_size() != 0) { in ComputeConstantGraph()
|
/external/tensorflow/tensorflow/compiler/aot/ |
D | compile.cc | 79 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()
|
D | codegen.cc | 175 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/ |
D | xrt_compile_ops.cc | 153 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/ |
D | xla_jit_compiled_cpu_function.cc | 112 arg_shapes.reserve(program_shape->parameters_size()); in Compile() 113 for (int i = 0; i < program_shape->parameters_size(); ++i) { in Compile()
|
D | xla_jit_compiled_cpu_function_test.cc | 221 ASSERT_EQ(program_shape.parameters_size(), 2); in TEST() 272 ASSERT_EQ(program_shape.parameters_size(), 2); in TEST()
|
D | xla_compiler_test.cc | 1292 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/ |
D | hlo_test_base.cc | 64 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/ |
D | tpu_compile_op_support.cc | 67 if (program_shape.parameters_size() != argument_shapes.size()) { in CreateModuleConfig() 69 program_shape.parameters_size(), in CreateModuleConfig()
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_embedding_optimization_parameters_utils.cc | 157 const int num_inputs = program_shape.parameters_size(); in GetBaseAuxiliaryParameterCount()
|
/external/tensorflow/tensorflow/compiler/xrt/tests/ |
D | raw_api_test.cc | 370 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/ |
D | tfcompile_test.cc | 642 ASSERT_EQ(muladd_shape->parameters_size(), 2); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.cc | 3691 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/ |
D | model.cc | 1533 int64 num_parameters = node_proto.parameters_size(); in FromProtoHelper()
|