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

  /external/tensorflow/tensorflow/contrib/lite/kernels/
kernel_util.cc 100 int64_t out_dims = std::max(dims1, dims2); local
102 TfLiteIntArrayCreate(out_dims), TfLiteIntArrayFree);
103 for (int i = 0; i < out_dims; ++i) {
107 shape->data[out_dims - i - 1] = std::max(d1, d2);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_fft_impl.h 72 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; local
73 out_dims[0] = input_batch;
77 out_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i];
85 output(out, out_dims);
98 output.device(device) = full_fft.slice(zero_start_indices, out_dims);
113 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; local
114 out_dims[0] = input_batch;
118 out_dims[i + 1] = fft_shape[i];
126 output(out, out_dims);
  /external/tensorflow/tensorflow/core/kernels/
diag_op.cc 80 const int out_dims = num_dims / 2; variable
85 for (int i = 0; i < out_dims; i++) {
87 context, tensor.dim_size(i) == tensor.dim_size(i + out_dims),
90 i, " and ", i + out_dims, " do not match."));
94 for (int i = 0; i < out_dims; ++i) {
mkl_conv_grad_filter_ops.cc 156 mkl_context.out_dims = out_backprop_in_mkl_format
272 int out_dims; member in struct:tensorflow::MklConv2DCustomBackpropFilterOp::__anon39839
314 CHECK_EQ(dnnLayoutCreate_F32(&lt_out_backprop, out_dims, out_sizes,
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
diag_op.cc 140 const int out_dims = num_dims / 2; variable
151 for (int i = 0; i < out_dims; i++) {
153 ctx, dims[i] == dims[i + out_dims],
155 ": dimensions ", i, " and ", i + out_dims,
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.h 256 std::vector<int64> out_dims = {rhs.n1(), rhs.n2(), rhs.n3()}; local
259 out_dims[i] = lhs_dims[i];
262 out_dims[i] = lhs_dims[i] + rhs_dims[i];
265 auto result = MakeUnique<Array3D<T>>(out_dims[0], out_dims[1], out_dims[2]);
290 std::vector<int64> out_dims = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; local
293 out_dims[i] = lhs_dims[i];
296 out_dims[i] = lhs_dims[i] + rhs_dims[i];
299 auto result = MakeUnique<Array4D<T>>(out_dims[0], out_dims[1], out_dims[2]
    [all...]

Completed in 461 milliseconds