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

  /external/tensorflow/tensorflow/core/kernels/
eigen_softmax_test.cc 39 Eigen::array<int, 1> depth_dim; local
40 depth_dim[0] = 0;
49 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
54 (reference.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
eigen_softmax.h 49 Eigen::array<typename internal::traits<Input>::Index, 1> depth_dim; local
50 depth_dim[0] = 0;
60 Eigen::IndexList<Eigen::type2index<0> > depth_dim;
71 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
76 (output.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
softmax_op_functor.h 60 Eigen::IndexList<Eigen::type2index<1> > depth_dim; local
  /external/eigen/unsupported/test/
cxx11_tensor_forced_eval.cpp 56 Eigen::array<int, 1> depth_dim; local
57 depth_dim[0] = 0;
65 Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
pooling_ops.cc 424 int depth_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); variable
425 int64 depth = out_backprop_shape.dim_size(depth_dim);
  /external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc 672 DimensionHandle depth_dim = c->Dim( local
690 {output_rows, output_cols}, depth_dim,
    [all...]

Completed in 151 milliseconds