| /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 | 1034 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
1061 keep_dims = deprecated_argument_lookup(
1062 "keepdims", keepdims, "keep_dims", keep_dims)
1063 if keep_dims is None:
1064 keep_dims = False
1085 m_ss = math_ops.reduce_sum(m_ss, axes, keepdims=keep_dims, name="mean_ss")
1086 v_ss = math_ops.reduce_sum(v_ss, axes, keepdims=keep_dims, name="var_ss")
1116 x=x, axes=axes, shift=shift, keep_dims=keepdims, name=name)
1157 keep_dims=None, argument
1186 keep_dims = deprecated_argument_lookup(
[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=[])
358 ["foo", "bar"], reduction_indices=0, keep_dims=True)
|
| /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_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_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_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_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_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_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
|
| /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 | 104 if (op_context->params->keep_dims) { in ResizeOutputTensor()
298 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
318 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
351 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
368 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
382 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
395 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
406 if (op_context.params->keep_dims && NumDimensions(input) == 4 && in EvalMean()
426 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
440 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMean()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseReduceMax.pbtxt | 24 name: "keep_dims"
71 name: "keep_dims"
120 name: "keep_dims"
170 name: "keep_dims"
|
| D | SparseReduceMaxSparse.pbtxt | 32 name: "keep_dims"
87 name: "keep_dims"
144 name: "keep_dims"
202 name: "keep_dims"
|
| D | SparseReduceSum.pbtxt | 24 name: "keep_dims"
76 name: "keep_dims"
130 name: "keep_dims"
185 name: "keep_dims"
|
| D | Mean.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| D | Sum.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| D | Max.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| D | Prod.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| D | Min.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| D | SparseReduceSumSparse.pbtxt | 32 name: "keep_dims"
92 name: "keep_dims"
154 name: "keep_dims"
217 name: "keep_dims"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 66 BoolAttr keep_dims; member
104 none_af, fc_format, keep_dims); in CreateLayerNorm()
112 none_af, fc_format, keep_dims); in CreateGate()
114 none_af, fc_format, keep_dims); in CreateGate()
189 lstm.projection_bias(), none_af, fc_format, keep_dims); in LoadForLSTMOp()
204 keep_dims = builder.getBoolAttr(false); in runOnFunction()
|