Searched refs:sizes_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 40 const TensorShape sizes_shape = ctx->InputShape(1); in Compile() local
42 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(sizes_shape), in Compile()
44 sizes_shape.DebugString())); in Compile()
45 const int64 num_dims = sizes_shape.num_elements(); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_proto_op.cc | 737 TensorShape sizes_shape = shape_prefix; in Compute() local
738 sizes_shape.AddDim(field_count); in Compute()
740 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, sizes_shape, &sizes_tensor)); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2155 ShapeHandle sizes_shape; in SliceShape() local
2156 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &sizes_shape)); in SliceShape()
2159 TF_RETURN_IF_ERROR(c->Merge(begin_shape, sizes_shape, &begin_shape)); in SliceShape()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 684 TensorShape sizes_shape({num_dims, 2}); in CreatePadNode() local
685 sizes_shape.AsProto(sizes_tensor_proto.mutable_tensor_shape()); in CreatePadNode()
990 TensorShape sizes_shape({shape.dims()}); in CreateSliceNode() local
991 sizes_shape.AsProto(sizes_tensor_proto.mutable_tensor_shape()); in CreateSliceNode()
|