Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc40 const TensorShape sizes_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Ddecode_proto_op.cc737 TensorShape sizes_shape = shape_prefix; in Compute() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc684 TensorShape sizes_shape({num_dims, 2}); in CreatePadNode() local
990 TensorShape sizes_shape({shape.dims()}); in CreateSliceNode() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc2155 ShapeHandle sizes_shape; in SliceShape() local