Searched refs:reduced_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scan_ops.cc | 73 int64 reduced_shape[3] = {1, 1, 1}; in Compute() local
75 reduced_shape[0] *= input.dim_size(i); in Compute()
77 reduced_shape[1] = input.dim_size(axis); in Compute()
79 reduced_shape[2] *= input.dim_size(i); in Compute()
82 functor::Scan<Device, Reducer, T>()(d, input.shaped<T, 3>(reduced_shape), in Compute()
83 output->shaped<T, 3>(reduced_shape), in Compute()
|
| D | sparse_reduce_op.cc | 43 TensorShape reduced_shape; member
102 reduction.reduced_shape = TensorShape(out_dim_sizes); in SparseTensorReduceHelper()
182 ctx, ctx->allocate_output(0, reduction.reduced_shape, &out_values)); in Compute()
279 0, TensorShape({nnz, reduction.reduced_shape.dims()}), in Compute()
315 2, TensorShape({reduction.reduced_shape.dims()}), in Compute()
318 auto out_dim_sizes = reduction.reduced_shape.dim_sizes(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | 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/ |
| D | sparse_grad.py | 113 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
|
| D | 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])
|
| D | math_ops.py | 2530 def reduced_shape(input_shape, axes): function
|
| D | metrics_impl.py | 2107 filled_selected_id_shape = math_ops.reduced_shape(ids_shape,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduction_ops_test.py | 59 output = math_ops.reduced_shape(shape, axes=axes)
|