OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:input_batch_dim
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc
39
auto
input_batch_dim
= dnums.input_batch_dimension();
local
60
new_input_dim_order[0] =
input_batch_dim
;
61
new_input_dims[0] = input->shape().dimensions(
input_batch_dim
);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc
81
auto
input_batch_dim
= conv_dnums.input_batch_dimension();
local
114
if (
input_batch_dim
== output_batch_dim &&
191
backward_conv_dnums.set_input_feature_dimension(
input_batch_dim
);
Completed in 106 milliseconds