Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc67 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
70 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P()
72 HloInstruction::CreateParameter(1, param_shape, "input")); in TEST_P()
74 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs, rhs)); in TEST_P()
82 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
85 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P()
87 HloInstruction::CreateParameter(1, param_shape, "input")); in TEST_P()
89 HloInstruction::CreateTranspose(param_shape, lhs, {1, 0})); in TEST_P()
91 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()
93 llvm_ir::IrArray param_x_array(param_x_val, param_shape); in TEST_F()
95 llvm_ir::IrArray::Index zero_2d({zero, zero}, param_shape, zero->getType()); in TEST_F()
Dcpu_vectorization_test.cc92 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
94 HloInstruction::CreateParameter(0, param_shape, "input0")); in TEST_P()
96 HloInstruction::CreateParameter(1, param_shape, "input1")); in TEST_P()
98 HloInstruction::CreateBinary(param_shape, spec.opcode, param0, param1)); in TEST_P()
Dcpu_intrinsic_test.cc95 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
97 HloInstruction::CreateParameter(0, param_shape, "input")); in TEST_P()
99 HloInstruction::CreateUnary(param_shape, spec.opcode, param)); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ftz_test.cc37 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule() local
40 /* parameter_number=*/0, param_shape, "x")); in CreateBinaryOpModule()
42 /* parameter_number=*/1, param_shape, "y")); in CreateBinaryOpModule()
43 builder.AddInstruction(HloInstruction::CreateBinary(param_shape, op, x, y)); in CreateBinaryOpModule()
54 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule() local
57 /* parameter_number=*/0, param_shape, "x")); in CreateUnaryOpModule()
58 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.cc129 auto param_shape = ShapeUtil::MakeShape(F32, {32, 32}); in TEST_F() local
132 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F()
136 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.cc315 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
318 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
320 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
336 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
342 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
369 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
372 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
388 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
392 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
410 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_normalization.cc106 const auto& param_shape = mean_shape; in Run() local
107 const int buffer_size = RequiredBufferSizeForShape(param_shape); in Run()
112 mul_param_array.copy_shape(param_shape); in Run()
113 add_param_array.copy_shape(param_shape); in Run()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py395 param_shape = (list(axis_to_dim.values())[0],)
398 param_shape = [
403 param_shape.insert(1, 1)
410 shape=param_shape,
421 1.0, dtype=self._param_dtype, shape=param_shape)
426 shape=param_shape,
437 0.0, dtype=self._param_dtype, shape=param_shape)
448 shape=param_shape,
458 shape=param_shape,
477 shape=param_shape,
[all …]
Dadvanced_activations.py148 param_shape = list(input_shape[1:])
151 param_shape[i - 1] = 1
153 shape=param_shape,
Dnormalization_test.py685 param_shape = [batch_input_shape[i] for i in axis]
687 for dim in param_shape:
689 beta = np.arange(param_elems, dtype='float64').reshape(param_shape)
690 gamma = np.arange(1, param_elems + 1, dtype='float64').reshape(param_shape)
743 param_shape = [batch_input_shape[i] for i in axis]
745 for dim in param_shape:
747 beta = np.arange(param_elems, dtype='float64').reshape(param_shape)
748 gamma = np.arange(1, param_elems + 1, dtype='float64').reshape(param_shape)
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc549 const HloInstruction& param, const Shape& param_shape, in CreateLiteralForConstrainedUses() argument
630 return MakeFakeLiteralInternalWithBounds(param_shape, engine, -1, in CreateLiteralForConstrainedUses()
635 return LiteralUtil::Zero(param_shape.element_type()); in CreateLiteralForConstrainedUses()
637 return LiteralUtil::One(param_shape.element_type()); in CreateLiteralForConstrainedUses()
641 return MakeFakeLiteralInternal(param_shape, engine, in CreateLiteralForConstrainedUses()
646 return MakeFakeLiteralInternal(param_shape, engine, no_duplicates, in CreateLiteralForConstrainedUses()
655 const Shape& param_shape, in MakeConstrainedArgument() argument
659 return CreateLiteralForConstrainedUses(constrained_uses, param, param_shape, in MakeConstrainedArgument()
689 const Shape& param_shape = (module_config.has_entry_computation_layout() && in MakeFakeArguments() local
699 arguments[i] = MakeConstrainedArgument(*dataflow, *params[i], param_shape, in MakeFakeArguments()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc63 const Shape& param_shape = param->shape(); in CodegenNestedComputation() local
65 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in CodegenNestedComputation()
67 llvm_ir::ByteSizeOf(param_shape, module_->getDataLayout()); in CodegenNestedComputation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc193 const Shape& param_shape = in Verify() local
196 TF_RET_CHECK(ShapeUtil::IndexIsValid(param_shape, alias.parameter_index)); in Verify()
200 ShapeUtil::GetSubshape(param_shape, alias.parameter_index); in Verify()
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()
Dlayout_assignment_test.cc720 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
728 param_shape)); in TEST_F()
861 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
865 param_shape)); in TEST_F()
902 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
906 param_shape)); in TEST_F()
Dmemory_space_assignment_test.cc894 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
898 HloInstruction::CreateParameter(1, param_shape, "p1")); in TEST_P()
902 HloInstruction::CreateBitcast(param_shape, negate)); in TEST_P()
904 HloInstruction::CreateBinary(param_shape, HloOpcode::kAdd, bitcast, p1)); in TEST_P()
923 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
927 HloInstruction::CreateParameter(1, param_shape, "p1")); in TEST_P()
962 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
966 HloInstruction::CreateParameter(1, param_shape, "p1")); in TEST_P()
1026 Shape param_shape = ShapeUtil::MakeShape(F32, {6}); in TEST_P() local
1045 HloInstruction::CreateParameter(1, param_shape, "p1")); in TEST_P()
[all …]
Dhlo_creation_utils.cc588 for (const Shape* param_shape : domain) { in CreateComputationWithSignature() local
590 param_idx, *param_shape, StrCat("param.", param_idx))); in CreateComputationWithSignature()
Dshape_inference.cc3198 const Shape& param_shape = to_apply.parameters(i); in InferCallShape() local
3199 if (!ShapeUtil::Compatible(arg_shape, param_shape)) { in InferCallShape()
3203 i, ShapeUtil::HumanString(param_shape), in InferCallShape()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc313 std::string param_shape = param_match[2]; in ParseBufferAssignment() local
314 TupleShape shape = TupleShapeFromString(param_shape); in ParseBufferAssignment()
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py694 param_shape = [2, 5]
695 expected_lookup_result_shape = [None] + param_shape
712 num_shards, vocab_size, shape=param_shape, dtype=dtype)
753 param_shape = [2, 3]
762 num_shards, vocab_size, shape=param_shape, dtype=dtype)
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc568 const Shape& param_shape = module->entry_computation() in TEST_F() local
572 EXPECT_TRUE(param_shape.is_dynamic_dimension(0)); in TEST_F()

12