Searched refs:desired_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/periodic_resample/python/kernel_tests/ |
| D | periodic_resample_op_test.py | 38 desired_shape = numpy.array([6, None])
43 result = periodic_resample(input_tensor, desired_shape).eval()
49 desired_shape = numpy.array([5, None])
54 result = periodic_resample(input_tensor, desired_shape).eval()
60 desired_shape = numpy.array([4, 4, None])
68 result = periodic_resample(input_tensor, desired_shape).eval()
78 desired_shape = numpy.array([4, 4, 4, None])
93 result = periodic_resample(input_tensor, desired_shape).eval()
109 desired_shape = numpy.array([4, 4, None])
114 output = periodic_resample(x, desired_shape)
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 209 const IndexVecT& desired_shape, in process_desired_shape() argument
217 if (desired_shape[i] < 1) { in process_desired_shape()
227 context, desired_shape[i] >= input_tensor_shape.dim_size(i), in process_desired_shape()
233 ", desired shape has size ", desired_shape[i], ".")); in process_desired_shape()
235 (*target_dimensions)[i] = desired_shape[i]; in process_desired_shape()
270 const tensorflow::PartialTensorShape& desired_shape, in do_periodic_resample_op() argument
276 OP_REQUIRES(context, rank == desired_shape.dims(), in do_periodic_resample_op()
280 desired_shape.dims(), ".")); in do_periodic_resample_op()
286 process_desired_shape(context, original_shape, desired_shape.dim_sizes(), in do_periodic_resample_op()
354 const tensorflow::PartialTensorShape& desired_shape) { in create_output_tensor() argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops.cc | 30 tensorflow::PartialTensorShape desired_shape; in __anon7c7f74300102() local
31 TF_RETURN_IF_ERROR(c->GetAttr("shape", &desired_shape)); in __anon7c7f74300102()
38 desired_shape, &result_shape_handle)); in __anon7c7f74300102()
45 if (desired_shape.dim_size(i) < 1) { in __anon7c7f74300102()
48 target_dimensions[i] = desired_shape.dim_size(i); in __anon7c7f74300102()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 583 absl::Span<HloInstruction*> instrs, const Shape& desired_shape, in UnflattenTupleInstr() argument
585 CHECK(desired_shape.IsTuple()) << ShapeUtil::HumanString(desired_shape); in UnflattenTupleInstr()
592 for (int64 i = 0; i < desired_shape.tuple_shapes_size(); ++i) { in UnflattenTupleInstr()
593 const Shape& subshape = desired_shape.tuple_shapes(i); in UnflattenTupleInstr()
611 desired_shape.tuple_shapes(i), new_instrs); in UnflattenTupleInstr()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1449 PartialTensorShape desired_shape; in __anondb9326b22302() local
1450 TF_RETURN_IF_ERROR(c->GetAttr("shape", &desired_shape)); in __anondb9326b22302()
1452 int rank = desired_shape.dims(); in __anondb9326b22302()
1457 desired_shape, &desired_shape_handle)); in __anondb9326b22302()
|