Home
last modified time | relevance | path

Searched refs:param_shape (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc65 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()
Dcpu_noalias_test.cc45 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()
Dcpu_intrinsic_test.cc80 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/
Dgpu_ftz_test.cc34 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()
Dgpu_noalias_test.cc37 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()
Dgpu_index_test.cc46 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()
Dgpu_ldg_test.cc114 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/
Dnn_batchnorm_test.py77 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/
Dcpu_instruction_fusion_test.cc313 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/
Dresolve_batch_normalization.cc104 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/
Dnormalization.py318 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 …]
Dadvanced_activations.py122 param_shape = list(input_shape[1:])
125 param_shape[i - 1] = 1
127 shape=param_shape,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc55 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/
Dhlo_input_output_alias_config.cc188 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()
Dlayout_assignment_test.cc723 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()
Dwhile_loop_invariant_code_motion_test.cc33 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()
Dhlo_creation_utils.cc451 for (const Shape* param_shape : domain) { in CreateComputationWithSignature() local
453 param_idx, *param_shape, StrCat("param.", param_idx))); in CreateComputationWithSignature()
Dshape_inference.cc2853 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()
Dhlo_parser.cc534 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/
Dembedding_ops_test.py710 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/
Dembedding_ops_test.py689 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/
Draw_api_test.cc805 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/
Dxla_builder_test.cc496 const Shape& param_shape = module->entry_computation() in TEST_F() local
500 EXPECT_TRUE(param_shape.is_dynamic_dimension(0)); in TEST_F()
Dxla_builder.cc265 Shape param_shape(instr.shape()); in SetDynamicBinding() local
266 Shape* param_shape_ptr = &param_shape; in SetDynamicBinding()
272 *instr.mutable_shape() = param_shape.ToProto(); in SetDynamicBinding()