| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 78 param_shape = [2]
80 m_val = np.random.random_sample(param_shape).astype(np.float32)
81 v_val = np.random.random_sample(param_shape).astype(np.float32)
82 beta_val = np.random.random_sample(param_shape).astype(np.float32)
83 gamma_val = np.random.random_sample(param_shape).astype(np.float32)
127 param_shape = [5]
130 m_val = np.random.random_sample(param_shape).astype(np.float64)
131 v_val = np.random.random_sample(param_shape).astype(np.float64)
132 beta_val = np.random.random_sample(param_shape).astype(np.float64)
133 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 | 249 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
252 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
256 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0)); in TEST_F()
272 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
278 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
302 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
305 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
321 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
325 HloInstruction::CreateParameter(0, param_shape, "param")); in TEST_F()
343 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
[all …]
|
| D | ir_emitter.cc | 1236 auto param_shape = parameter->shape(); in HandleParameter() local
1266 param_address_untyped, IrShapeType(param_shape)->getPointerTo()); in HandleParameter()
1269 if (!ShapeUtil::IsOpaque(param_shape)) { in HandleParameter()
1270 AttachAlignmentMetadataForLoad(param_address_untyped, param_shape); in HandleParameter()
1271 AttachDereferenceableMetadataForLoad(param_address_untyped, param_shape); in HandleParameter()
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | normalization.py | 260 param_shape = (list(axis_to_dim.values())[0],)
263 param_shape = [axis_to_dim[i] if i in axis_to_dim
267 param_shape.insert(1, 1)
274 shape=param_shape,
284 1.0, dtype=param_dtype, shape=param_shape)
289 shape=param_shape,
299 0.0, dtype=param_dtype, shape=param_shape)
310 shape=param_shape,
317 shape=param_shape,
344 self.renorm_mean = _renorm_variable('renorm_mean', param_shape)
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | resolve_batch_normalization.cc | 88 const auto& param_shape = mean_shape; in Run() local
89 const int buffer_size = RequiredBufferSizeForShape(param_shape); in Run()
94 mul_param_array.copy_shape(param_shape); in Run()
95 add_param_array.copy_shape(param_shape); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 54 const Shape& param_shape = param->shape(); in EmitBasePointersForNestedComputation() local
56 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in EmitBasePointersForNestedComputation()
58 llvm_ir::ByteSizeOf(param_shape, module_->getDataLayout()); in EmitBasePointersForNestedComputation()
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| D | advanced_activations.py | 123 param_shape = list(input_shape[1:])
124 self.param_broadcast = [False] * len(param_shape)
127 param_shape[i - 1] = 1
130 shape=param_shape,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 697 param_shape = [2, 5]
698 expected_lookup_result_shape = [None] + param_shape
714 num_shards, vocab_size, shape=param_shape, dtype=dtype)
744 param_shape = [2, 3]
753 num_shards, vocab_size, shape=param_shape, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion_test.cc | 32 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
50 const Shape& param_shape, HloModule* module) { in MakeAlwaysTrueComputation() argument
53 HloInstruction::CreateParameter(0, param_shape, "param")); in MakeAlwaysTrueComputation()
|
| D | while_loop_simplifier_test.cc | 35 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
98 const Shape& param_shape, HloModule* module) { in MakeAlwaysTrueComputation() argument
101 HloInstruction::CreateParameter(0, param_shape, "param")); in MakeAlwaysTrueComputation()
|
| D | layout_assignment_test.cc | 619 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
627 param_shape)); in TEST_F()
|
| D | shape_inference.cc | 2438 const Shape& param_shape = to_apply.parameters(i); in InferCallShape() local
2439 if (!ShapeUtil::Compatible(arg_shape, param_shape)) { in InferCallShape()
2443 i, ShapeUtil::HumanString(param_shape).c_str(), in InferCallShape()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 652 param_shape = [2, 5]
653 expected_lookup_result_shape = [None] + param_shape
669 num_shards, vocab_size, shape=param_shape, dtype=dtype)
698 param_shape = [2, 3]
707 num_shards, vocab_size, shape=param_shape, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 305 const Shape& param_shape = computation->parameter_instruction(p)->shape(); in ParseComputations() local
306 if (param_shape.has_layout()) { in ParseComputations()
309 ->ResetLayout(param_shape.layout()); in ParseComputations()
|