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()
|