/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_eigen_dot_operation_test.cc | 65 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local 68 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P() 70 HloInstruction::CreateParameter(1, param_shape, "input")); in TEST_P() 72 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs, rhs)); in TEST_P() 80 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local 83 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P() 85 HloInstruction::CreateParameter(1, param_shape, "input")); in TEST_P() 87 HloInstruction::CreateTranspose(param_shape, lhs, {1, 0})); in TEST_P() 89 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs_transposed, rhs)); in TEST_P()
|
D | cpu_noalias_test.cc | 45 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 47 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F() 49 HloInstruction::CreateParameter(1, param_shape, "y")); in TEST_F() 92 llvm_ir::IrArray param_x_array(param_x_val, param_shape); in TEST_F()
|
D | cpu_intrinsic_test.cc | 80 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local 82 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P() 84 HloInstruction::CreateUnary(param_shape, spec.opcode, param)); in TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | gpu_ftz_test.cc | 34 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule() local 37 /* parameter_number=*/0, param_shape, "x")); in CreateBinaryOpModule() 39 /* parameter_number=*/1, param_shape, "y")); in CreateBinaryOpModule() 40 builder.AddInstruction(HloInstruction::CreateBinary(param_shape, op, x, y)); in CreateBinaryOpModule() 51 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule() local 54 /* parameter_number=*/0, param_shape, "x")); in CreateUnaryOpModule() 55 builder.AddInstruction(HloInstruction::CreateUnary(param_shape, op, x)); in CreateUnaryOpModule()
|
D | gpu_noalias_test.cc | 37 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 39 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F() 41 HloInstruction::CreateParameter(1, param_shape, "y")); in TEST_F()
|
D | gpu_index_test.cc | 46 auto param_shape = ShapeUtil::MakeShape(F32, {5, 7, 2}); in TEST_F() local 48 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F() 50 HloInstruction::CreateParameter(1, param_shape, "y")); in TEST_F()
|
D | gpu_ldg_test.cc | 114 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 117 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F() 121 param_shape, HloOpcode::kAdd, param, param)), in TEST_F()
|
/external/tensorflow/tensorflow/python/ops/ |
D | nn_batchnorm_test.py | 77 param_shape = [2] 79 m_val = np.random.random_sample(param_shape).astype(np.float32) 80 v_val = np.random.random_sample(param_shape).astype(np.float32) 81 beta_val = np.random.random_sample(param_shape).astype(np.float32) 82 gamma_val = np.random.random_sample(param_shape).astype(np.float32) 126 param_shape = [5] 129 m_val = np.random.random_sample(param_shape).astype(np.float64) 130 v_val = np.random.random_sample(param_shape).astype(np.float64) 131 beta_val = np.random.random_sample(param_shape).astype(np.float64) 132 gamma_val = np.random.random_sample(param_shape).astype(np.float64) [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_instruction_fusion_test.cc | 313 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local 316 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F() 318 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F() 334 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 340 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F() 367 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 370 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F() 386 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local 390 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F() 408 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local [all …]
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_batch_normalization.cc | 104 const auto& param_shape = mean_shape; in Run() local 105 const int buffer_size = RequiredBufferSizeForShape(param_shape); in Run() 110 mul_param_array.copy_shape(param_shape); in Run() 111 add_param_array.copy_shape(param_shape); in Run()
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | normalization.py | 318 param_shape = (list(axis_to_dim.values())[0],) 321 param_shape = [axis_to_dim[i] if i in axis_to_dim 325 param_shape.insert(1, 1) 332 shape=param_shape, 343 1.0, dtype=self._param_dtype, shape=param_shape) 348 shape=param_shape, 359 0.0, dtype=self._param_dtype, shape=param_shape) 370 shape=param_shape, 380 shape=param_shape, 411 self.renorm_mean = _renorm_variable('renorm_mean', param_shape) [all …]
|
D | advanced_activations.py | 122 param_shape = list(input_shape[1:]) 125 param_shape[i - 1] = 1 127 shape=param_shape,
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_nested.cc | 55 const Shape& param_shape = param->shape(); in EmitBasePointersForNestedComputation() local 57 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in EmitBasePointersForNestedComputation() 59 llvm_ir::ByteSizeOf(param_shape, module_->getDataLayout()); in EmitBasePointersForNestedComputation()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_input_output_alias_config.cc | 188 const Shape& param_shape = in Verify() local 191 TF_RET_CHECK(ShapeUtil::IndexIsValid(param_shape, alias.parameter_index)); in Verify() 195 ShapeUtil::GetSubshape(param_shape, alias.parameter_index); in Verify()
|
D | layout_assignment_test.cc | 723 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local 731 param_shape)); in TEST_F() 864 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local 868 param_shape)); in TEST_F() 908 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local 912 param_shape)); in TEST_F()
|
D | while_loop_invariant_code_motion_test.cc | 33 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape, 51 const Shape& param_shape, HloModule* module) { in MakeAlwaysTrueComputation() argument 54 HloInstruction::CreateParameter(0, param_shape, "param")); in MakeAlwaysTrueComputation()
|
D | hlo_creation_utils.cc | 451 for (const Shape* param_shape : domain) { in CreateComputationWithSignature() local 453 param_idx, *param_shape, StrCat("param.", param_idx))); in CreateComputationWithSignature()
|
D | shape_inference.cc | 2853 const Shape& param_shape = to_apply.parameters(i); in InferCallShape() local 2854 if (!ShapeUtil::Compatible(arg_shape, param_shape)) { in InferCallShape() 2858 i, ShapeUtil::HumanString(param_shape), in InferCallShape()
|
D | hlo_parser.cc | 534 const Shape& param_shape = computation->parameter_instruction(p)->shape(); in ParseComputations() local 537 ->CopyLayoutFromShape(param_shape)); in ParseComputations()
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
D | embedding_ops_test.py | 710 param_shape = [2, 5] 711 expected_lookup_result_shape = param_shape 727 num_shards, vocab_size, shape=param_shape, dtype=dtype) 757 param_shape = [2, 3] 766 num_shards, vocab_size, shape=param_shape, dtype=dtype)
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | embedding_ops_test.py | 689 param_shape = [2, 5] 690 expected_lookup_result_shape = [None] + param_shape 707 num_shards, vocab_size, shape=param_shape, dtype=dtype) 748 param_shape = [2, 3] 757 num_shards, vocab_size, shape=param_shape, dtype=dtype)
|
/external/tensorflow/tensorflow/compiler/xrt/tests/ |
D | raw_api_test.cc | 805 auto param_shape = in TEST() local 807 auto param = xla::Parameter(&builder, 0, param_shape, "param"); in TEST() 821 *shapes->add_parameters() = param_shape.ToProto(); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 496 const Shape& param_shape = module->entry_computation() in TEST_F() local 500 EXPECT_TRUE(param_shape.is_dynamic_dimension(0)); in TEST_F()
|
D | xla_builder.cc | 265 Shape param_shape(instr.shape()); in SetDynamicBinding() local 266 Shape* param_shape_ptr = ¶m_shape; in SetDynamicBinding() 272 *instr.mutable_shape() = param_shape.ToProto(); in SetDynamicBinding()
|