| /external/tensorflow/tensorflow/core/kernels/ |
| scan_ops.cc | 73 int64 reduced_shape[3] = {1, 1, 1}; variable
75 reduced_shape[0] *= input.dim_size(i);
77 reduced_shape[1] = input.dim_size(axis);
79 reduced_shape[2] *= input.dim_size(i);
82 functor::Scan<Device, Reducer, T>()(d, input.shaped<T, 3>(reduced_shape),
83 output->shaped<T, 3>(reduced_shape),
|
| sparse_reduce_op.cc | 43 TensorShape reduced_shape; member in struct:tensorflow::ReduceDetails
53 // // Set output shape to reduction.reduced_shape.
102 reduction.reduced_shape = TensorShape(out_dim_sizes);
182 ctx, ctx->allocate_output(0, reduction.reduced_shape, &out_values));
279 0, TensorShape({nnz, reduction.reduced_shape.dims()}),
315 2, TensorShape({reduction.reduced_shape.dims()}),
318 auto out_dim_sizes = reduction.reduced_shape.dim_sizes();
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| normalization_test.py | 120 reduced_shape = (input_shape[0], input_shape[axis])
121 expected_mean = np.zeros(reduced_shape)
122 expected_var = np.ones(reduced_shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_grad.py | 113 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
|
| math_grad.py | 61 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
70 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
129 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
[all...] |
| math_ops.py | 2530 def reduced_shape(input_shape, axes): function
[all...] |
| metrics_impl.py | [all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| reduction_ops_test.py | 59 output = math_ops.reduced_shape(shape, axes=axes)
72 """Check that reduced_shape does the right thing with zero dimensions."""
[all...] |
