| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_architectures_test.py | 43 def get_test_data(self, input_shape, target_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 415 target_shape, argument
554 target_shape, argument
715 target_shape, argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 78 TensorShape target_shape; in QueryMeta() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 109 auto target_shape = target_shape_in.vec<int64>(); in ReshapeSparseTensor() local
|
| D | deserialize_sparse_string_op.cc | 201 Tensor target_shape(DT_INT64, TensorShape({ndims + output.dims() - 2})); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 110 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_reshape_op_test.py | 336 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 340 HloInstruction* SliceValidData(HloInstruction* hlo, const Shape& target_shape, in SliceValidData()
|
| D | spmd_partitioner_util.cc | 1795 HloSharding CreateMatchingShardingOnDims(const Shape& target_shape, in CreateMatchingShardingOnDims()
|
| D | spmd_partitioner.cc | 869 Shape target_shape = shard_shape; in ReplicatePartial() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2096 Shape target_shape = fusion->shape(); in HandleFusion() local
3026 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitThreadLocalBufferPointer()
3076 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitGlobalBufferPointer()
3101 const Shape& target_shape) { in EmitBufferPointer()
3114 const Shape& target_shape = op->shape(); in EmitTargetAddressForOp() local
3134 const Shape& target_shape = target_op->shape(); in EmitTargetElementLoop() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 2775 def test_dtype(image, target_shape, target_method): argument
3101 target_shape, argument
3319 def test_dtype(image, target_shape, target_method): argument
3637 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 483 def __init__(self, target_shape, **kwargs): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 842 Output target_shape = ops::Concat( in TEST_F() local
888 Output target_shape = in TEST_F() local
920 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
952 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 298 const Shape& target_shape, in ReshapeSharding()
|
| D | layout_assignment.cc | 812 const Shape& target_shape = in CreateCopyWithNewLayout() local
|