| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_nested.cc | 54 const Shape& param_shape = param->shape(); local
56 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo());
58 llvm_ir::ByteSizeOf(param_shape, module_->getDataLayout());
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_batch_normalization.cc | 88 const auto& param_shape = mean_shape; local
89 const int buffer_size = RequiredBufferSizeForShape(param_shape);
94 mul_param_array.copy_shape(param_shape);
95 add_param_array.copy_shape(param_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 249 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); local
252 HloInstruction::CreateParameter(0, param_shape, "param"));
256 HloInstruction::CreateUnary(param_shape, HloOpcode::kExp, param0));
272 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); local
278 HloInstruction::CreateParameter(0, param_shape, "param"));
302 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); local
305 HloInstruction::CreateParameter(0, param_shape, "param"));
321 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); local
325 HloInstruction::CreateParameter(0, param_shape, "param"));
343 Shape param_shape = ShapeUtil::MakeShape(F32, {4}) local
361 Shape param_shape = ShapeUtil::MakeShape(F32, {4, 4}); local
380 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); local
398 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); local
417 Shape param_shape = ShapeUtil::MakeShape(F32, {3, 4}); local
[all...] |
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| 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/python/keras/_impl/keras/layers/ |
| 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/python/layers/ |
| 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/layers/python/layers/ |
| 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/ |
| while_loop_invariant_code_motion_test.cc | 32 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
50 const Shape& param_shape, HloModule* module) {
53 HloInstruction::CreateParameter(0, param_shape, "param"));
|
| while_loop_simplifier_test.cc | 35 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
98 const Shape& param_shape, HloModule* module) {
101 HloInstruction::CreateParameter(0, param_shape, "param"));
|
| layout_assignment_test.cc | 619 Shape param_shape = ShapeUtil::MakeTupleShape( local
627 param_shape));
[all...] |
| shape_inference.cc | 2438 const Shape& param_shape = to_apply.parameters(i); local
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 305 const Shape& param_shape = computation->parameter_instruction(p)->shape(); local
306 if (param_shape.has_layout()) {
309 ->ResetLayout(param_shape.layout());
[all...] |
