Searched refs:dest_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 119 const Shape& source_shape, const Shape& dest_shape, in StrideConfig() argument
128 dimensions[LayoutUtil::Minor(dest_shape.layout(), 0)]) { in StrideConfig()
130 dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension); in StrideConfig()
132 minor_dimension = LayoutUtil::Minor(dest_shape.layout(), 0); in StrideConfig()
438 absl::Span<const NativeT> src, const Shape& dest_shape, in CopyElementsBetween() argument
440 CHECK(ShapeUtil::Compatible(dest_shape, src_shape)); in CopyElementsBetween()
441 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
444 std::vector<int64> index(dest_shape.rank()); in CopyElementsBetween()
446 dest[IndexUtil::MultidimensionalIndexToLinearIndex(dest_shape, index)] = in CopyElementsBetween()
448 } while (IndexUtil::BumpIndices(dest_shape, absl::MakeSpan(index))); in CopyElementsBetween()
[all …]
|
| D | literal.h | 258 StatusOr<Literal> ConvertToShape(const Shape& dest_shape) const;
736 StrideConfig(const Shape& source_shape, const Shape& dest_shape,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1269 TensorShape dest_shape; in ComputeWithReservedSpace() local
1276 dest_shape = ShapeFromFormat(tensor_format_, in_batch, in ComputeWithReservedSpace()
1278 OP_REQUIRES(context, x.CopyFrom(x, dest_shape), in ComputeWithReservedSpace()
1300 auto alloc_shape = use_reshape ? dest_shape : x_shape; in ComputeWithReservedSpace()
1430 TensorShape dest_shape; in ComputeWithReservedSpace() local
1437 dest_shape = ShapeFromFormat(tensor_format_, in_batch, in ComputeWithReservedSpace()
1439 OP_REQUIRES(context, x.CopyFrom(x, dest_shape), in ComputeWithReservedSpace()
1441 OP_REQUIRES(context, y_backprop.CopyFrom(y_backprop, dest_shape), in ComputeWithReservedSpace()
1446 auto alloc_shape = use_reshape ? dest_shape : x_shape; in ComputeWithReservedSpace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 993 const Shape& dest_shape = ShapeUtil::GetTupleElementShape(operand_shape, 0); in HandleCopyDone() local
995 if (!ShapesSame(dest_shape, src_shape, in HandleCopyDone()
1001 StringifyShape(dest_shape), StringifyShape(src_shape), in HandleCopyDone()
|
| D | algebraic_simplifier.cc | 241 const Shape& dest_shape = convert->shape(); in IsConvertPairNoOp() local
245 const PrimitiveType dest_type = dest_shape.element_type(); in IsConvertPairNoOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3638 const Shape dest_shape = TypeToShape(dest.getType()); in TryBuildConstantInitializerThunk() local
3640 BuildConstantInitializerThunk(literal_bytes, dest_slice, dest_shape); in TryBuildConstantInitializerThunk()
3672 const Shape dest_shape = TypeToShape(dest.getType()); in BuildInitializerThunkForMlir() local
3674 dest_shape, ir_emitter_context_->gpu_device_info()); in BuildInitializerThunkForMlir()
3716 const Shape dest_shape = TypeToShape(dest.getType()); in BuildFusedInitializerThunkForMlir() local
3718 dest_shape, ir_emitter_context_->gpu_device_info()); in BuildFusedInitializerThunkForMlir()
|