Home
last modified time | relevance | path

Searched refs:reduced_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscan_ops.cc73 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()
Dsparse_reduce_op.cc43 TensorShape reduced_shape; member
102 reduction.reduced_shape = TensorShape(out_dim_sizes); in SparseTensorReduceHelper()
184 ctx, ctx->allocate_output(0, reduction.reduced_shape, &out_values)); in Compute()
283 0, TensorShape({nnz, reduction.reduced_shape.dims()}), in Compute()
319 2, TensorShape({reduction.reduced_shape.dims()}), in Compute()
322 auto out_dim_sizes = reduction.reduced_shape.dim_sizes(); in Compute()
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dnormalization_test.py120 reduced_shape = (input_shape[0], input_shape[axis])
121 expected_mean = np.zeros(reduced_shape)
122 expected_var = np.ones(reduced_shape)
334 reduced_shape = []
337 reduced_shape.append(a)
349 expected_mean = np.zeros(reduced_shape)
350 expected_var = np.ones(reduced_shape)
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py111 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
Dmath_grad.py85 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
94 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
153 output_shape_kept_dims = math_ops.reduced_shape(input_shape, op.inputs[1])
Dmath_ops.py3174 def reduced_shape(input_shape, axes): function
Dmetrics_impl.py2229 filled_selected_id_shape = math_ops.reduced_shape(ids_shape,
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduction_ops_test.py61 output = math_ops.reduced_shape(shape, axes=axes)