| /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 | 1168 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
1209 keep_dims = deprecated_argument_lookup(
1210 "keepdims", keepdims, "keep_dims", keep_dims)
1211 if keep_dims is None:
1212 keep_dims = False
1236 m_ss = math_ops.reduce_sum(m_ss, axes, keepdims=keep_dims, name="mean_ss")
1237 v_ss = math_ops.reduce_sum(v_ss, axes, keepdims=keep_dims, name="var_ss")
1268 x=x, axes=axes, shift=shift, keep_dims=keepdims, name=name)
1311 keep_dims=None, argument
1340 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=[])
356 ["foo", "bar"], reduction_indices=0, keep_dims=True)
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reduce.cc | 163 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
176 params->keep_dims, temp_index, resolved_axis, in EvalMean()
182 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
200 params->keep_dims, temp_index, resolved_axis, temp_buffer)); in EvalMean()
212 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
218 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
236 num_axis, params->keep_dims, temp_index, in EvalMean()
249 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
285 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
302 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 174 Value input_value, ElementsAttr axes_elems, bool keep_dims);
179 Value input_value, ElementsAttr axes_elems, bool keep_dims);
184 Value input_value, ElementsAttr axes_elems, bool keep_dims);
189 Value input_value, ElementsAttr axes_elems, bool keep_dims);
194 Value input_value, ElementsAttr axes_elems, bool keep_dims);
199 Value input_value, ElementsAttr axes_elems, bool keep_dims);
204 Value input_value, ElementsAttr axes_elems, bool keep_dims);
|
| /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_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_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_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_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_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"]]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseReduceMax.pbtxt | 24 name: "keep_dims"
71 name: "keep_dims"
120 name: "keep_dims"
170 name: "keep_dims"
|
| D | Min.pbtxt | 16 name: "keep_dims"
71 name: "keep_dims"
128 name: "keep_dims"
186 name: "keep_dims"
244 name: "keep_dims"
|
| D | Max.pbtxt | 16 name: "keep_dims"
71 name: "keep_dims"
128 name: "keep_dims"
186 name: "keep_dims"
244 name: "keep_dims"
|
| D | SparseReduceSum.pbtxt | 24 name: "keep_dims"
76 name: "keep_dims"
130 name: "keep_dims"
185 name: "keep_dims"
|
| D | SparseReduceMaxSparse.pbtxt | 32 name: "keep_dims"
87 name: "keep_dims"
144 name: "keep_dims"
202 name: "keep_dims"
|
| /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 | Sum.pbtxt | 16 name: "keep_dims"
73 name: "keep_dims"
132 name: "keep_dims"
192 name: "keep_dims"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 67 BoolAttr keep_dims; member
105 none_af, fc_format, keep_dims); in CreateLayerNorm()
113 none_af, fc_format, keep_dims); in CreateGate()
115 none_af, fc_format, keep_dims); in CreateGate()
190 lstm.projection_bias(), none_af, fc_format, keep_dims); in LoadForLSTMOp()
205 keep_dims = builder.getBoolAttr(false); in runOnFunction()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 84 std::initializer_list<int> axis, bool keep_dims) { in MeanOpConstModel() argument
89 CreateReducerOptions(builder_, keep_dims).Union()); in MeanOpConstModel()
98 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel() argument
103 CreateReducerOptions(builder_, keep_dims).Union()); in MeanOpDynamicModel()
113 std::initializer_list<int> axis, bool keep_dims) { in SumOpConstModel() argument
118 CreateReducerOptions(builder_, keep_dims).Union()); in SumOpConstModel()
127 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel() argument
132 CreateReducerOptions(builder_, keep_dims).Union()); in SumOpDynamicModel()
142 std::initializer_list<int> axis, bool keep_dims) { in ProdOpConstModel() argument
147 CreateReducerOptions(builder_, keep_dims).Union()); in ProdOpConstModel()
[all …]
|