| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| H A D | graph_transferer.cc | 51 const char KEEP_DIMS_ATTR_NAME[] = "keep_dims";
691 bool keep_dims = false; local
693 if (context->GetAttr(KEEP_DIMS_ATTR_NAME, &keep_dims).ok()) {
694 padding_id = keep_dims ? Padding::SAME : Padding::VALID;
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| H A D | import_tensorflow.cc | 748 if (HasAttr(node, "keep_dims")) {
749 op->keep_dims = GetBoolAttr(node, "keep_dims");
1126 if (HasAttr(node, "keep_dims")) {
1127 op->keep_dims = GetBoolAttr(node, "keep_dims");
1141 if (HasAttr(node, "keep_dims")) {
1142 op->keep_dims = GetBoolAttr(node, "keep_dims");
1485 if (HasAttr(node, "keep_dims")) {
[all...] |
| H A D | export_tensorflow.cc | 1510 if (src_op.keep_dims) {
1511 (*new_op->mutable_attr())["keep_dims"].set_b(true);
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| H A D | ops_test.py | 663 self.original_lt.tensor, 1, keep_dims=True),
671 self.original_lt.tensor, 1, keep_dims=True),
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 588 const bool keep_dims = local
589 node->attr().count("keep_dims") && node->attr().at("keep_dims").b();
595 if (!IsReshape(*fanout) && !keep_dims) {
597 // of shape [], [1], [1, 1], [1, 1, ...]. If keep_dims isn't true, we
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| H A D | gbdt_batch_test.py | 46 math_ops.square(predictions - label), 1, keep_dims=True)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| H A D | embedding_ops.py | 473 ids, math_ops.reduce_prod(shape, keep_dims=True))
|
| H A D | layers.py | 774 mean, variance = nn.moments(inputs, moments_axes, keep_dims=True)
2173 mean, variance = nn.moments(inputs, norm_axes, keep_dims=True)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| H A D | math_utils.py | 63 maximum = math_ops.reduce_max(diagonal, axis=-1, keep_dims=True)
|
| /external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_fused_batchnorm_test.py | 101 math_ops.cast(x, scale.dtype), [0, 1, 2], keep_dims=False)
|
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| H A D | schema_generated.h | 3301 bool keep_dims; member in struct:tflite::MeanOptionsT
3303 : keep_dims(false) {
3312 bool keep_dims() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
3328 void add_keep_dims(bool keep_dims) { argument
3329 fbb_.AddElement<uint8_t>(MeanOptions::VT_KEEP_DIMS, static_cast<uint8_t>(keep_dims), 0);
3345 bool keep_dims = false) {
3347 builder_.add_keep_dims(keep_dims);
5311 { auto _e = keep_dims(); _o->keep_dims = _e; };
5322 auto _keep_dims = _o->keep_dims;
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_constructor_test.cc | 2898 key: "keep_dims"
2982 key: "keep_dims"
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| H A D | generate_examples.py | 709 "keep_dims": [True, False],
720 "keep_dims": [True, False],
743 input_tensor, axis=axis, keep_dims=parameters["keep_dims"])
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| H A D | metric_ops.py | 3226 ], keep_dims=True)
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 2801 const int num_axis_dimensions, bool keep_dims, int* temp_index,
|