Searched defs:target_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 80 TensorShape target_shape; in Query() local
137 TensorShape target_shape; in QueryMeta() local
|
| /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/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 94 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 413 target_shape, argument
521 target_shape, argument
|
| /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()
2895 const BufferAllocation::Slice& slice, const Shape& target_shape) { in EmitGlobalBufferPointer()
2920 const Shape& target_shape) { in EmitBufferPointer()
2933 const Shape& target_shape = op->shape(); in EmitTargetAddressForOp() local
2953 const Shape& target_shape = target_op->shape(); in EmitTargetElementLoop() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 406 def __init__(self, target_shape, **kwargs): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 902 Output target_shape = ops::Concat( in TEST_F() local
941 Output target_shape = in TEST_F() local
973 Output target_shape = ops::Const(s, {4, 3, 28, 28}, {4}); in TEST_F() local
1005 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 | 785 const Shape& target_shape = in CreateCopyWithNewLayout() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 2672 target_shape, argument
3184 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
|