OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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(<_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