Home
last modified time | relevance | path

Searched defs:out_sizes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dget_item_op.cc56 std::vector<int64> out_sizes = {}; in Compile() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc124 const std::vector<int64> out_sizes = { in Compute() local
Dextract_image_patches_op.cc98 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops.cc138 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, depth}; in Compute() local