| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 80 TensorShape target_shape; in Query() local
82 s = slice.SliceTensorShape(shape_, &target_shape); in Query()
87 int64 total_size = target_shape.num_elements(); in Query()
137 TensorShape target_shape; in QueryMeta() local
139 s = slice.SliceTensorShape(shape_, &target_shape); in QueryMeta()
144 int64 total_size = target_shape.num_elements(); in QueryMeta()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1014 target_shape = array_ops.shape(target)
1058 [target_shape, array_ops.shape(out)[1:]], axis=0)
1121 target_shape = target.shape
1122 if target_shape.rank is None:
1125 dim = target_shape.dims[0]
1126 if not (target_shape.with_rank_at_least(2) and
1132 if target_shape.is_fully_defined():
1133 batch_size = int(target_shape[0])
1134 target_row_size = target_shape.num_elements() // batch_size
1136 target_shape = array_ops.shape(target)
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 94 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement() argument
103 if (!target_shape.IsOpaque()) { in EmitGetTupleElement()
106 ByteSizeOf(target_shape, src_buffer->getModule()->getDataLayout())); in EmitGetTupleElement()
110 llvm::Type* element_type = ShapeToIrType(target_shape, module); in EmitGetTupleElement()
|
| D | tuple_ops.h | 80 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 199 Tensor target_shape(DT_INT64, TensorShape({ndims + output.dims() - 2})); in Compute() local
201 target_shape.vec<int64>()(i) = serialized_sparse.shape().dim_size(i); in Compute()
204 target_shape.vec<int64>()(i + ndims - 1) = output.shape().data()[i + 1]; in Compute()
207 Reshape(context, output.indices(), input_shape, target_shape, in Compute()
|
| D | reshape_util.cc | 61 auto target_shape = target_shape_in.vec<int64>(); in Reshape() local
63 const int64 size = target_shape(d); in Reshape()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 406 def __init__(self, target_shape, **kwargs): argument
408 self.target_shape = tuple(target_shape)
457 output_shape += tuple(s if s != -1 else None for s in self.target_shape)
461 self.target_shape)
466 (array_ops.shape(inputs)[0],) + self.target_shape)
469 config = {'target_shape': self.target_shape}
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2181 Shape target_shape = fusion->shape(); in HandleFusion() local
2836 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitThreadLocalBufferPointer() argument
2842 attr_builder.addAlignmentAttr(MinimumAlignmentForShape(target_shape)); in EmitThreadLocalBufferPointer()
2843 attr_builder.addDereferenceableAttr(ByteSizeOf(target_shape)); in EmitThreadLocalBufferPointer()
2865 if (!target_shape.IsOpaque()) { in EmitThreadLocalBufferPointer()
2866 AttachAlignmentMetadataForLoad(param_address_untyped, target_shape); in EmitThreadLocalBufferPointer()
2868 target_shape); in EmitThreadLocalBufferPointer()
2884 &b_, MinimumAlignmentForShape(target_shape)); in EmitThreadLocalBufferPointer()
2891 return BitCast(tempbuf_address, IrShapeType(target_shape)->getPointerTo()); in EmitThreadLocalBufferPointer()
2895 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitGlobalBufferPointer() argument
[all …]
|
| D | ir_emitter.h | 263 const Shape& target_shape);
267 const BufferAllocation::Slice& slice, const Shape& target_shape);
271 const Shape& target_shape);
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column.py | 125 target_shape = [shape[0], shape[1], num_elements]
127 array_ops.reshape(dense_tensor, shape=target_shape))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 413 target_shape, argument
422 image, target_shape, align_corners=align_corners)
521 target_shape, argument
530 image, target_shape, align_corners=align_corners)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| D | mnist_eager_multigpu.py | 53 target_shape=[28, 28, 1],
|
| D | mnist_tf1_tpu.py | 51 target_shape=[28, 28, 1],
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2178 target_shape = []
2181 target_shape.append(1)
2183 target_shape.append(array_ops.shape(x)[axis])
2184 target_shape = array_ops.stack(target_shape)
2186 broadcast_mean = array_ops.reshape(mean, target_shape)
2187 broadcast_var = array_ops.reshape(var, target_shape)
2191 broadcast_gamma = array_ops.reshape(gamma, target_shape)
2195 broadcast_beta = array_ops.reshape(beta, target_shape)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-reshape.pbtxt | 89 argspec: "args=[\'self\', \'target_shape\'], varargs=None, keywords=kwargs, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-reshape.pbtxt | 89 argspec: "args=[\'self\', \'target_shape\'], varargs=None, keywords=kwargs, defaults=None"
|
| /external/tensorflow/tensorflow/examples/tf2_showcase/ |
| D | mnist.py | 92 target_shape=input_shape,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 2384 for target_shape in target_shapes:
2385 y = image_ops.resize_images_v2(image, target_shape, method)
2436 for target_shape, img_shape in zip(target_shapes, img_shapes):
2447 expected = np.array(expected_data).reshape(target_shape)
2672 target_shape, argument
2677 target_height, target_width = target_shape
2879 for target_shape in target_shapes:
2880 y = image_ops.resize_images(image, target_shape, method)
2882 target_shape == image.shape[1:3]):
2932 for target_shape, img_shape in zip(target_shapes, img_shapes):
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 375 target_shape = self._target_shape(array_ops.shape(tensor), num_elements)
376 return array_ops.reshape(tensor, shape=target_shape)
4472 target_shape = [shape[0], shape[1], math_ops.reduce_prod(shape[2:])]
4473 id_tensor = sparse_ops.sparse_reshape(id_tensor, target_shape)
4475 weight_tensor = sparse_ops.sparse_reshape(weight_tensor, target_shape)
|
| D | feature_column.py | 3178 target_shape = [shape[0], shape[1], math_ops.reduce_prod(shape[2:])]
3179 id_tensor = sparse_ops.sparse_reshape(id_tensor, target_shape)
3181 weight_tensor = sparse_ops.sparse_reshape(weight_tensor, target_shape)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 902 Output target_shape = ops::Concat( in TEST_F() local
905 Output reshape = ops::Reshape(s, inputs, target_shape); in TEST_F()
941 Output target_shape = in TEST_F() local
945 Output reshape = ops::Reshape(s, inputs, target_shape); in TEST_F()
973 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
974 Output reshape = ops::Reshape(s, inputs, target_shape); in TEST_F()
1005 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
1006 Output reshape = ops::Reshape(s, inputs, target_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 785 const Shape& target_shape = in CreateCopyWithNewLayout() local
792 if (ShapeUtil::Equal(target_shape, instr_shape)) { in CreateCopyWithNewLayout()
799 CreateCopyWithNewLayout(target_shape, gte)); in CreateCopyWithNewLayout()
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy_test.py | 236 l.Reshape(target_shape=input_shape, input_shape=(28 * 28,)),
|