| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_architectures_test.py | 42 def get_test_data(self, input_shape, target_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 414 target_shape, argument
543 target_shape, argument
704 target_shape, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 61 auto target_shape = target_shape_in.vec<int64>(); in Reshape() local
|
| D | deserialize_sparse_string_op.cc | 199 Tensor target_shape(DT_INT64, TensorShape({ndims + output.dims() - 2})); in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 78 TensorShape target_shape; in QueryMeta() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 109 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2424 Shape target_shape = fusion->shape(); in HandleFusion() local
3155 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitThreadLocalBufferPointer()
3205 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitGlobalBufferPointer()
3230 const Shape& target_shape) { in EmitBufferPointer()
3243 const Shape& target_shape = op->shape(); in EmitTargetAddressForOp() local
3263 const Shape& target_shape = target_op->shape(); in EmitTargetElementLoop() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 436 def __init__(self, target_shape, **kwargs): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 945 Output target_shape = ops::Concat( in TEST_F() local
984 Output target_shape = in TEST_F() local
1016 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
1048 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 811 const Shape& target_shape = in CreateCopyWithNewLayout() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 2740 target_shape, argument
3252 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
|