HomeSort by relevance Sort by last modified time
    Searched full:dims2d (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
eigen_softmax_test.cc 44 Tensor<float, 2>::Dimensions dims2d; local
45 dims2d[0] = 1;
46 dims2d[1] = batch;
49 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
54 (reference.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
eigen_softmax.h 54 DSizes<typename internal::traits<Input>::Index, 2> dims2d; local
55 dims2d[0] = 1;
56 dims2d[1] = dimensions(input)[1];
65 dims2d;
66 dims2d.set(1, dimensions(input)[1]);
71 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) *
76 (output.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
  /external/eigen/unsupported/test/
cxx11_tensor_forced_eval.cpp 58 Tensor<float, 2>::Dimensions dims2d; local
59 dims2d[0] = 1;
60 dims2d[1] = 3;
65 Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
README.md 321 Tensor<...> Y = ((X - X.maximum(depth_dim).reshape(dims2d).broadcast(bcast))
329 ((X - X.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast))
338 Y = Y / (Y.sum(depth_dim).reshape(dims2d).broadcast(bcast));
344 Y = Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
355 (Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast))).eval();
    [all...]

Completed in 104 milliseconds