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

  /external/tensorflow/tensorflow/compiler/tf2xla/lib/
batch_dot.h 30 // setting the `transpose_x` or `transpose_y` flag to `true`. Similarly, each
41 // c_o = r_y if transpose_y else c_y
48 xla::ComputationDataHandle y, bool transpose_x, bool transpose_y,
batch_dot.cc 30 xla::ComputationDataHandle y, bool transpose_x, bool transpose_y,
65 int y_inner_dim = transpose_y ? (ndims - 1) : (ndims - 2);
72 " transpose: ", transpose_y);
83 int y_outer_dim = transpose_y ? (ndims - 2) : (ndims - 1);
103 auto rhs = transpose_y ? builder->Transpose(y, {1, 0}) : y;
  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding_test.cc 64 HloInstruction* transpose_y = local
72 /*rhs=*/transpose_y, dot_dnums));
175 HloInstruction* transpose_y = local
183 /*rhs=*/transpose_y, dot_dnums));
190 {transpose_y, dot}, "outlined", entry_computation);
222 HloInstruction* transpose_y = local
235 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i)));
238 x->shape(), transpose_y->shape(), window, dnums);
241 conv_shape.ValueOrDie(), x, transpose_y, window, dnums));
275 HloInstruction* transpose_y local
    [all...]

Completed in 217 milliseconds