| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 65 std::vector<const xla::Shape*> arg_layout_ptrs(pshape->parameters_size()); in CompileXla()
66 std::vector<xla::Shape> arg_layouts(pshape->parameters_size()); in CompileXla()
67 for (int i = 0; i < pshape->parameters_size(); ++i) { in CompileXla()
|
| D | codegen.cc | 170 size_t num_args = ps.parameters_size(); in GenArgMethods()
253 size_t num_args = ps.parameters_size(); in GenVariableMethods()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 161 ASSERT_EQ(program_shape.parameters_size(), in TEST_F()
162 program_shape_copy.parameters_size()); in TEST_F()
163 for (int i = 0; i < program_shape.parameters_size(); ++i) { in TEST_F()
|
| D | shape.cc | 209 std::vector<string> parameter_strings(parameters_size()); in ToString()
210 for (int i = 0; i < parameters_size(); ++i) { in ToString()
|
| D | shape.h | 239 int parameters_size() const { return parameters_.size(); } in parameters_size() function
|
| /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 | 98 computation.proto().host_program_shape().parameters_size()); in CompileExecutable()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 115 config.program_shape().parameters_size()); in Compile()
117 config.program_shape().parameters_size()); in Compile()
118 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 | 105 arg_shapes.reserve(program_shape->parameters_size()); in Compile()
106 for (int i = 0; i < program_shape->parameters_size(); ++i) { in Compile()
|
| D | xla_jit_compiled_cpu_function_test.cc | 121 ASSERT_EQ(program_shape.parameters_size(), 2); in TEST()
|
| D | xla_compiler_test.cc | 1224 ASSERT_EQ(program_shape->parameters_size(), 2); in TEST_F()
1294 ASSERT_EQ(program_shape->parameters_size(), 2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | local_service.cc | 128 if (argument_layouts.size() != program_shape.parameters_size()) { in CompileExecutable()
131 program_shape.parameters_size(), argument_layouts.size()); in CompileExecutable()
|
| D | shape_inference.cc | 64 if (reducer_shape.parameters_size() != inputs * 2) { in VerifyReducerShape()
68 inputs * 2, reducer_shape.parameters_size()); in VerifyReducerShape()
1127 if (arg_shapes.size() != to_apply.parameters_size()) { in InferMapShape()
1131 to_apply.parameters_size(), arg_shapes.size()); in InferMapShape()
1142 for (int i = 0; i < to_apply.parameters_size(); ++i) { in InferMapShape()
2132 if (select_shape.parameters_size() != 2) { in InferSelectAndScatterShape()
2136 select_shape.parameters_size()); in InferSelectAndScatterShape()
2523 if (condition.parameters_size() != 1) { in InferWhileShape()
2525 condition.parameters_size()); in InferWhileShape()
2527 if (body.parameters_size() != 1) { in InferWhileShape()
[all …]
|
| D | hlo_module.cc | 283 TF_RET_CHECK(expected_program_shape.parameters_size() == in CreateFromProto()
285 for (int i = 0; i < expected_program_shape.parameters_size(); ++i) { in CreateFromProto()
|
| D | service.cc | 271 if (program_shape.parameters_size() != argument_shapes.size()) { in CreateModuleConfig()
273 program_shape.parameters_size(), in CreateModuleConfig()
1081 if (arg->computation().host_program_shape().parameters_size() != 0) { in ComputeConstantGraph()
|
| /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/compiler/xla/python/ |
| D | numpy_bridge.cc | 157 Safe_PyObjectPtr arg_shapes = make_safe(PyTuple_New(shape.parameters_size())); in PyProgramShapeInfoFromXlaProgramShape()
158 for (int i = 0; i < shape.parameters_size(); ++i) { in PyProgramShapeInfoFromXlaProgramShape()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 234 for (int64 i = 0; i < input_program_shape.parameters_size(); ++i) { in XlaCompiledProgramShape()
739 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
794 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
840 EXPECT_EQ(program_shape.parameters_size(), 1); in TEST()
1233 EXPECT_EQ(program_shape.parameters_size(), 2); in TEST()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 576 ASSERT_EQ(muladd_shape->parameters_size(), 2); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 399 if (alias.param_number >= program_shape.parameters_size()) { in PopulateInputOutputAlias()
402 program_shape.parameters_size()); in PopulateInputOutputAlias()
|