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

  /external/tensorflow/tensorflow/cc/tutorials/
example_trainer.cc 90 auto y_flat = y.flat<float>(); local
94 Eigen::Tensor<float, 0, Eigen::RowMajor> lambda = (x_flat * y_flat).sum();
96 lambda(), x_flat(0), x_flat(1), y_flat(0), y_flat(1));
  /external/tensorflow/tensorflow/core/distributed_runtime/
rpcbench_test.cc 161 auto y_flat = y.flat<float>(); local
165 x_flat(1), y_flat(0), y_flat(1));
  /external/tensorflow/tensorflow/compiler/xla/tests/
dot_operation_test.cc 559 auto y_flat = Reshape(y, {0, 1, 2, 3}, {4, 2, 2}); local
567 auto y_slice = Slice(y_flat, {i, 0, 0}, {i + 1, 2, 2}, {1, 1, 1});
    [all...]

Completed in 242 milliseconds