OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:padded_shape
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/data/
padded_batch_dataset_op.cc
57
PartialTensorShape
padded_shape
;
variable
60
padded_shape_t.NumElements(), &
padded_shape
));
61
padded_shapes.push_back(std::move(
padded_shape
));
239
const PartialTensorShape&
padded_shape
=
variable
242
for (int dim = 0; dim <
padded_shape
.dims(); ++dim) {
243
if (
padded_shape
.dim_size(dim) == -1) {
246
batch_component_shape.AddDim(
padded_shape
.dim_size(dim));
255
if (element_shape.dims() !=
padded_shape
.dims()) {
259
component_index, ": expected rank ",
padded_shape
.dims(),
262
for (int dim = 0; dim <
padded_shape
.dims(); ++dim)
[
all
...]
tensor_queue_dataset_op.cc
183
//
padded_shape
[r] is known.
186
//
padded_shape
[r] is unknown, find the maximum across
516
PartialTensorShape
padded_shape
;
variable
519
padded_shape_t.NumElements(), &
padded_shape
));
520
padded_shapes.push_back(std::move(
padded_shape
));
/external/tensorflow/tensorflow/core/kernels/
conv_ops_fused.cc
711
TensorShape
padded_shape
variable
[
all
...]
Completed in 95 milliseconds