HomeSort by relevance Sort by last modified time
    Searched refs:KeepDims (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
mean_test.cc 89 TEST(ConstMeanOpTest, KeepDims) {
117 TEST(DynamicMeanOpTest, KeepDims) {
  /external/tensorflow/tensorflow/tools/graph_transforms/
quantize_nodes_test.cc 528 dequantize_reduction_dims, Min::Attrs().KeepDims(false));
532 dequantize_reduction_dims, Max::Attrs().KeepDims(false));
627 dequantize_reduction_dims, Min::Attrs().KeepDims(false));
631 dequantize_reduction_dims, Max::Attrs().KeepDims(false));
753 dequantize_reduction_dims, Min::Attrs().KeepDims(false));
757 dequantize_reduction_dims, Max::Attrs().KeepDims(false));
    [all...]
  /external/tensorflow/tensorflow/cc/gradients/
nn_grad.cc 54 auto sum = Sum(scope, grad_inputs[0], {1}, Sum::KeepDims(true));
  /external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 450 bool KeepDims(const Operator& op) {
477 const bool keep_dims = KeepDims(*op);
    [all...]

Completed in 111 milliseconds