| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_eigen_dot_operation_test.cc | 67 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
82 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
|
| D | cpu_vectorization_test.cc | 92 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
|
| D | cpu_intrinsic_test.cc | 95 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
|
| D | cpu_noalias_test.cc | 45 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ftz_test.cc | 37 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule() local
54 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule() local
|
| D | gpu_noalias_test.cc | 37 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
| D | gpu_ldg_test.cc | 129 auto param_shape = ShapeUtil::MakeShape(F32, {32, 32}); in TEST_F() local
|
| D | gpu_index_test.cc | 46 auto param_shape = ShapeUtil::MakeShape(F32, {5, 7, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 315 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
336 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
369 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
388 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
410 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
428 Shape param_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
447 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
465 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
484 Shape param_shape = ShapeUtil::MakeShape(F32, {3, 4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_batch_normalization.cc | 106 const auto& param_shape = mean_shape; in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 549 const HloInstruction& param, const Shape& param_shape, in CreateLiteralForConstrainedUses()
655 const Shape& param_shape, in MakeConstrainedArgument()
689 const Shape& param_shape = (module_config.has_entry_computation_layout() && in MakeFakeArguments() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 63 const Shape& param_shape = param->shape(); in CodegenNestedComputation() local
|
| D | ir_emitter_unnested.cc | 4862 const Shape& param_shape = context.operand_shapes[id]; in EmitHlo021Tile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 193 const Shape& param_shape = in Verify() local
|
| D | layout_assignment_test.cc | 720 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
861 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
902 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
|
| D | memory_space_assignment_test.cc | 894 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
923 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
962 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
1026 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
1138 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
1177 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
1243 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
|
| D | while_loop_invariant_code_motion_test.cc | 51 const Shape& param_shape, HloModule* module) { in MakeAlwaysTrueComputation()
|
| D | hlo_creation_utils.cc | 588 for (const Shape* param_shape : domain) { in CreateComputationWithSignature() local
|
| D | shape_inference.cc | 3198 const Shape& param_shape = to_apply.parameters(i); in InferCallShape() local
|
| D | hlo_parser.cc | 857 const Shape& param_shape = computation->parameter_instruction(p)->shape(); in ParseComputations() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | driver.cc | 313 std::string param_shape = param_match[2]; in ParseBufferAssignment() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 303 def _testBatchNormArbitraryShapes(self, x_shape, param_shape, atol=0.0001, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 568 const Shape& param_shape = module->entry_computation() in TEST_F() local
|
| D | xla_builder.cc | 387 Shape param_shape(instr.shape()); in SetDynamicBinding() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1509 auto param_shape = in TEST() local
|