| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 354 def _npSuffStats(self, x, axes, shift, keep_dims): argument
357 m_ss = np.sum(x - shift, axis=axis, keepdims=keep_dims)
358 v_ss = np.sum((x - shift) * (x - shift), axis=axis, keepdims=keep_dims)
360 m_ss = np.sum(x, axis=axis, keepdims=keep_dims)
361 v_ss = np.sum(x * x, axis=axis, keepdims=keep_dims)
366 if not keep_dims:
370 def _opSuffStats(self, x, axes, shift, keep_dims): argument
371 return nn_impl.sufficient_statistics(x, axes, shift, keep_dims)
373 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
375 np_c, np_m, np_v, np_s = self._npSuffStats(x_val, axes, shift, keep_dims)
[all …]
|
| D | nn_impl.py | 811 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
838 keep_dims = deprecated_argument_lookup(
839 "keepdims", keepdims, "keep_dims", keep_dims)
840 if keep_dims is None:
841 keep_dims = False
861 m_ss = math_ops.reduce_sum(m_ss, axes, keepdims=keep_dims, name="mean_ss")
862 v_ss = math_ops.reduce_sum(v_ss, axes, keepdims=keep_dims, name="var_ss")
892 x=x, axes=axes, shift=shift, keep_dims=keepdims, name=name)
933 keep_dims=None, argument
962 keep_dims = deprecated_argument_lookup(
[all …]
|
| D | string_ops.py | 330 keep_dims=False, argument
335 keep_dims = deprecation.deprecated_argument_lookup(
336 "keepdims", keepdims, "keep_dims", keep_dims)
343 keep_dims=keep_dims,
356 inputs, axis, keep_dims=keepdims, separator=separator, name=name)
|
| D | math_ops.py | 1320 keep_dims=None): argument
1363 "keep_dims", keep_dims)
1467 keep_dims=None, argument
1520 "keep_dims", keep_dims)
1605 keep_dims=None): argument
1658 "keep_dims", keep_dims)
1858 keep_dims=None): argument
1889 "keep_dims", keep_dims)
1901 keep_dims=None): argument
1932 "keep_dims", keep_dims)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_join_op_test.py | 105 keep_dims=False, argument
121 keep_dims=keep_dims,
142 inputs=input_array, axis=axis, keep_dims=False, separator=separator)
144 inputs=input_array, axis=axis, keep_dims=True, separator=separator)
149 inputs=truth, axis=index, keep_dims=True, separator=separator)
275 keep_dims=True)
281 keep_dims=True)
287 keep_dims=True, axis=None)
293 input_array, expected_val, expected_shape, keep_dims=True, axis=[])
|
| D | sparse_ops_test.py | 614 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
620 np_ans = np.sum(np_ans, keepdims=keep_dims)
622 np_ans = np.max(np_ans, keepdims=keep_dims)
633 np_ans = np.sum(np_ans, axis=ra, keepdims=keep_dims)
635 np_ans = np.max(np_ans, axis=ra, keepdims=keep_dims)
640 keep_dims)
643 keep_dims)
649 keep_dims)
653 keep_dims)
732 def _testSparseReduceShape(self, sp_t, reduction_axes, ndims, keep_dims, argument
[all …]
|
| D | norm_op_test.py | 113 for keep_dims in False, True:
116 keep_dims, use_static_shape)
118 _GetNormOpTest(dtype, shape, ord, axis, keep_dims,
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 62 std::initializer_list<int> axis, bool keep_dims) { in MeanOpConstModel() argument
67 CreateReducerOptions(builder_, keep_dims).Union()); in MeanOpConstModel()
76 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel() argument
81 CreateReducerOptions(builder_, keep_dims).Union()); in MeanOpDynamicModel()
91 std::initializer_list<int> axis, bool keep_dims) { in SumOpConstModel() argument
96 CreateReducerOptions(builder_, keep_dims).Union()); in SumOpConstModel()
105 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel() argument
110 CreateReducerOptions(builder_, keep_dims).Union()); in SumOpDynamicModel()
120 std::initializer_list<int> axis, bool keep_dims) { in ProdOpConstModel() argument
125 CreateReducerOptions(builder_, keep_dims).Union()); in ProdOpConstModel()
[all …]
|
| D | reduce.cc | 101 if (op_context->params->keep_dims) { in ResizeOutputTensor()
293 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
338 op_context.params->keep_dims, GetTensorData<int>(temp_index), \ in EvalMean()
349 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
385 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
426 op_context->params->keep_dims, GetTensorData<int>(temp_index), in EvalLogic()
553 num_axis, op_context.params->keep_dims, in EvalSum()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EuclideanNorm.pbtxt | 27 name: "keep_dims"
35 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
36 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Mean.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Max.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Any.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Min.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Sum.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_Prod.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_All.pbtxt | 30 name: "keep_dims"
38 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
39 `reduction_indices`. If `keep_dims` is true, the reduced dimensions are
|
| D | api_def_ReduceJoin.pbtxt | 21 set to `1` depending on `keep_dims`.
25 name: "keep_dims"
53 tf.reduce_join(a, 0, keep_dims=True) ==> [["ac", "bd"]]
54 tf.reduce_join(a, 1, keep_dims=True) ==> [["ab"], ["cd"]]
|
| D | api_def_SparseReduceSumSparse.pbtxt | 29 name: "keep_dims"
41 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
42 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
|
| D | api_def_SparseReduceMaxSparse.pbtxt | 29 name: "keep_dims"
41 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
42 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
|
| D | api_def_SparseReduceMax.pbtxt | 35 name: "keep_dims"
47 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
48 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
|
| D | api_def_SparseReduceSum.pbtxt | 35 name: "keep_dims"
47 `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in
48 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/ |
| D | monte_carlo_impl.py | 199 axis=0, keep_dims=False, name=None): argument
331 return math_ops.reduce_mean(f(samples), axis=axis, keepdims=keep_dims)
351 return math_ops.reduce_mean(fx, axis=axis, keepdims=keep_dims)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 65 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
69 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
73 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
77 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
|
| D | tensorflow.math.pbtxt | 105 …, \'axis\', \'keepdims\', \'dtype\', \'name\', \'reduction_indices\', \'keep_dims\', \'input\'], v…
305 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
309 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
317 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
321 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
325 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
329 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
333 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
341 …ut_tensor\', \'axis\', \'keepdims\', \'name\', \'reduction_indices\', \'keep_dims\'], varargs=None…
|