| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | image_ops_test.py | 409 target_shape, argument
548 target_shape, argument
663 target_shape, argument
824 target_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 2208 Shape target_shape = fusion->shape(); in HandleFusion() local
3227 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitThreadLocalBufferPointer()
3278 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitGlobalBufferPointer()
3304 const Shape& target_shape) { in EmitBufferPointer()
3317 const Shape& target_shape = op->shape(); in EmitTargetAddressForOp() local
3337 const Shape& target_shape = target_op->shape(); in EmitTargetElementLoop() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | tensor_aten_ops.cpp | 28 const torch::lazy::Shape& target_shape) { in MaybeExpand()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 2093 target_shape, argument
2146 def target_shape(self): member in _Broadcaster
2267 def _get_broadcaster(source_shape, target_shape): argument
|
| H A D | dynamic_ragged_shape_test.py | 2397 def testBroadcasterInitRaises(self, source_shape, target_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_slice_set.cc | 78 TensorShape target_shape; in QueryMeta() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_split_with_sizes_copy.cpp | 91 ArrayRef<Tensor::SizesType> target_shape( in split_with_sizes_copy_out() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | reshape_util.cc | 115 auto target_shape = target_shape_in.vec<int64_t>(); in ReshapeSparseTensor() local
|
| H A D | deserialize_sparse_string_op.cc | 201 Tensor target_shape(DT_INT64, TensorShape({ndims + output.dims() - 2})); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | tuple_ops.cc | 88 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64_t index, in EmitGetTupleElement()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | tile.cc | 163 nvinfer1::ITensor *target_shape = nullptr; in Convert() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_reshape_op_test.py | 354 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | fft_handler.cc | 338 HloInstruction* SliceValidData(HloInstruction* hlo, const Shape& target_shape, in SliceValidData()
|
| H A D | spmd_partitioner_util.cc | 1670 const Shape& target_shape, const HloSharding& source_sharding, in CreateMatchingShardingOnDims()
|
| H A D | spmd_partitioner.cc | 1182 Shape target_shape = shard_shape; in ReplicatePartial() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_ops_test.py | 2992 def test_dtype(image, target_shape, target_method): argument
3317 target_shape, argument
3543 def test_dtype(image, target_shape, target_method): argument
3860 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | core.py | 476 def __init__(self, target_shape, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer_test.cc | 1270 Output target_shape = ops::Concat( in TEST_F() local
1316 Output target_shape = in TEST_F() local
1348 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
1380 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment.cc | 1002 const Shape& target_shape = in CreateCopyWithNewLayout() local
|
| H A D | hlo_sharding_util.cc | 356 const Shape& target_shape, in ReshapeSharding()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 513 c10::ShapeSymbol& target_shape = shape_vector[i]; in ComputeShapeFromReshape() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 188 std::vector<c10::SymInt> target_shape(total_dims, 0); in restore_reduced_dims() local
|