Searched refs:src_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 578 const TensorShape& src_shape, in ShapeFromFormat() argument
581 return src_shape; in ShapeFromFormat()
584 const int64 batch = GetTensorDim(src_shape, src_format, 'N'); in ShapeFromFormat()
585 const int64 channels = GetTensorDim(src_shape, src_format, 'C') * in ShapeFromFormat()
588 GetTensorSpatialDims(src_shape.dims(), src_format); in ShapeFromFormat()
592 gtl::ArraySlice<int64>(src_shape.dim_sizes())[GetTensorSpatialDimIndex( in ShapeFromFormat()
593 src_shape.dims(), src_format, spatial_dim)]; in ShapeFromFormat()
604 const TensorShape& src_shape, in ShapeFromFilterFormat() argument
607 return src_shape; in ShapeFromFilterFormat()
610 const int64 output_channels = GetFilterDim(src_shape, src_filter_format, 'O'); in ShapeFromFilterFormat()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 491 src_shape = RaggedTensorDynamicShape.from_tensor(rt_input)
492 src_shape = src_shape.broadcast_to_rank(dst_shape.rank)
522 if not src_shape.is_ragged(axis) and not dst_shape.is_ragged(axis):
523 src_size = src_shape.dimension_size(axis)
542 if not src_shape.is_ragged(axis) and dst_shape.is_ragged(axis):
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 459 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape() local
461 if (src_context->Value(src_context->Rank(src_shape)) == 0) { in ConstantPartialShape()
486 TF_RETURN_IF_ERROR(src_context->WithRank(src_shape, 1, &src_shape)); in ConstantPartialShape()
489 if (src_context->Value(src_context->Dim(src_shape, 0)) == 0) { in ConstantPartialShape()
544 evaluated ? &t : nullptr, src_shape, result)); in ConstantPartialShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 373 const Shape& src_shape) { in CopyElementsBetween() argument
374 CHECK(ShapeUtil::Compatible(dest_shape, src_shape)); in CopyElementsBetween()
381 src[IndexUtil::MultidimensionalIndexToLinearIndex(src_shape, index)]; in CopyElementsBetween()
2281 const auto& src_shape = shape_->tuple_shapes(i); in BorrowingLiteral() local
2282 CHECK(src_shape.IsArray()); in BorrowingLiteral()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 2181 void copy_shape(const Shape& src_shape) { *mutable_shape() = src_shape; }
|