Home
last modified time | relevance | path

Searched refs:shape_t (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc106 Status ValidateInputs(const Tensor *shape_t, const Tensor *reduction_axes_t) { in ValidateInputs() argument
108 if (!TensorShapeUtils::IsVector(shape_t->shape())) { in ValidateInputs()
111 shape_t->shape().DebugString()); in ValidateInputs()
123 if (axis < -shape_t->NumElements() || axis >= shape_t->NumElements()) { in ValidateInputs()
126 shape_t->NumElements(), " dimensions."); in ValidateInputs()
161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
164 OP_REQUIRES_OK(ctx, ctx->input("input_shape", &shape_t)); in Compute()
167 OP_REQUIRES_OK(ctx, ValidateInputs(shape_t, reduction_axes_t)); in Compute()
174 const auto shape_vec = shape_t->vec<int64>(); in Compute()
257 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
[all …]
Dparameterized_truncated_normal_op_test.cc29 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormal() local
30 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormal()
46 .Input(test::graph::Constant(g, shape_t)) in PTruncatedNormal()
58 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormal2SD() local
59 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormal2SD()
73 .Input(test::graph::Constant(g, shape_t)) in PTruncatedNormal2SD()
85 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormalOneTail() local
86 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormalOneTail()
100 .Input(test::graph::Constant(g, shape_t)) in PTruncatedNormalOneTail()
Dsparse_softmax_op.cc43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
46 OP_REQUIRES_OK(context, context->input("sp_shape", &shape_t)); in Compute()
55 TensorShapeUtils::IsVector(shape_t->shape()), in Compute()
60 shape_t->shape().DebugString())); in Compute()
61 OP_REQUIRES(context, shape_t->NumElements() >= 2, in Compute()
64 shape_t->SummarizeValue(3))); in Compute()
76 TensorShape(shape_t->flat<int64>()), &st)); in Compute()
Drandom_crop_op.cc39 const Tensor& shape_t = context->input(1); in Compute() local
40 OP_REQUIRES(context, shape_t.dims() == 1, in Compute()
42 shape_t.shape().DebugString())); in Compute()
43 OP_REQUIRES(context, shape_t.NumElements() == 2, in Compute()
45 shape_t.shape().DebugString())); in Compute()
47 auto shape_vec = shape_t.vec<int64>(); in Compute()
Dimage_resizer_state.h87 const Tensor& shape_t = context->input(1); in ValidateAndCalculateOutputSize() local
88 OP_REQUIRES(context, shape_t.dims() == 1, in ValidateAndCalculateOutputSize()
90 shape_t.shape().DebugString())); in ValidateAndCalculateOutputSize()
91 OP_REQUIRES(context, shape_t.NumElements() == 2, in ValidateAndCalculateOutputSize()
93 shape_t.shape().DebugString())); in ValidateAndCalculateOutputSize()
94 auto Svec = shape_t.vec<int32>(); in ValidateAndCalculateOutputSize()
Dsparse_dense_binary_op_shared.cc62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
65 OP_REQUIRES_OK(ctx, ctx->input("sp_shape", &shape_t)); in Compute()
75 TensorShapeUtils::IsVector(shape_t->shape()), in Compute()
80 shape_t->shape().DebugString())); in Compute()
83 const auto shape_vec = shape_t->vec<int64>(); in Compute()
Dresize_nearest_neighbor_op.cc191 const Tensor& shape_t = context->input(1); in Compute() local
192 OP_REQUIRES(context, shape_t.dims() == 1, in Compute()
194 shape_t.shape().DebugString())); in Compute()
195 OP_REQUIRES(context, shape_t.NumElements() == 2, in Compute()
197 shape_t.shape().DebugString())); in Compute()
199 auto sizes = shape_t.vec<int32>(); in Compute()
Dstateful_random_ops.cc105 const Tensor& shape_t = ctx->input(shape_input_idx); in ComputeImpl() local
107 OP_REQUIRES_OK(ctx, ctx->op_kernel().MakeShape(shape_t, &shape)); in ComputeImpl()
154 const Tensor& shape_t = ctx->input(0); in Compute() local
156 OP_REQUIRES_OK(ctx, ctx->op_kernel().MakeShape(shape_t, &shape)); in Compute()
Drandom_op.cc275 const Tensor& shape_t = ctx->input(0); in Compute() local
279 TensorShapeUtils::IsVector(shape_t.shape()) && in Compute()
280 (shape_t.dtype() == DataType::DT_INT32 || in Compute()
281 shape_t.dtype() == DataType::DT_INT64), in Compute()
284 shape_t.DebugString())); in Compute()
286 if (shape_t.dtype() == DataType::DT_INT32) { in Compute()
287 auto vec = shape_t.flat<int32>(); in Compute()
290 } else if (shape_t.dtype() == DataType::DT_INT64) { in Compute()
291 auto vec = shape_t.flat<int64>(); in Compute()
Dresize_bicubic_op_test.cc260 auto shape_t = shape.flat<int32>(); in ResizeBicubic() local
261 shape_t(0) = scale_y * size; in ResizeBicubic()
262 shape_t(1) = scale_x * size; in ResizeBicubic()
Dctc_decoder_ops.cc146 auto shape_t = p_shape->vec<int64>(); in StoreAllDecodedSequences() local
162 shape_t(0) = batch_size; in StoreAllDecodedSequences()
163 shape_t(1) = max_decoded; in StoreAllDecodedSequences()
Dexample_parsing_ops.cc690 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
691 shape_t(0) = num_elements; in Compute()
706 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
707 shape_t(0) = 0; in Compute()
816 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
817 shape_t(0) = feature_list_size; in Compute()
818 shape_t(1) = max_num_features; in Compute()
Dstateless_random_ops.cc74 const Tensor& shape_t = context->input(0); in Compute() local
77 OP_REQUIRES_OK(context, MakeShape(shape_t, &shape)); in Compute()
Drandom_poisson_op.cc289 const Tensor& shape_t = ctx->input(0); in Compute() local
293 OP_REQUIRES_OK(ctx, MakeShape(shape_t, &samples_shape)); in Compute()
Dnccl_ops.cc233 const Tensor& shape_t = c->input(0); in ComputeAsync() local
236 c, TensorShapeUtils::MakeShape(shape_t.vec<int32>(), &shape), done); in ComputeAsync()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dimage_ops.cc88 const Tensor& shape_t = ctx->input(2); in Compute() local
89 OP_REQUIRES(ctx, shape_t.dims() == 1, in Compute()
91 shape_t.shape().DebugString())); in Compute()
92 OP_REQUIRES(ctx, shape_t.NumElements() == 2, in Compute()
94 shape_t.shape().DebugString())); in Compute()
95 auto shape_vec = shape_t.vec<int32>(); in Compute()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc388 auto shape_t = (*output_sparse_shapes_tensor)[d].vec<int64>(); in BatchExampleProtoToTensors() local
389 shape_t(0) = batch_size; in BatchExampleProtoToTensors()
390 shape_t(1) = sparse_tensor_batch_shapes.max_num_features; in BatchExampleProtoToTensors()
/external/tensorflow/tensorflow/core/ops/
Dnn_ops_test.cc536 Tensor shape_t = test::AsTensor<int32>(shape); in TEST() local
537 op.input_tensors[0] = &shape_t; in TEST()
Darray_ops_test.cc527 Tensor shape_t(DT_INT64, TensorShape{3}); in TEST() local
528 test::FillValues<int64>(&shape_t, {2, 10, 3}); in TEST()
529 op.input_tensors[1] = &shape_t; in TEST()
/external/python/cpython3/Lib/test/
Dtest_buffer.py1325 shape_t = [randrange(2, 10) for _ in range(ndim)]
1326 nitems = prod(shape_t)
1327 for shape in permutations(shape_t):
1537 shape_t = (2, 3, 5)
1538 nitems = prod(shape_t)
1539 for shape in permutations(shape_t):
1696 shape_t = (2, 3, 5)
1697 ndim = len(shape_t)
1698 nitems = prod(shape_t)
1699 for shape in permutations(shape_t):
[all …]
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py129 shape_t = array_ops.shape(c)
130 self.assertAllEqual([10000, 3000], sess.run(shape_t))
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dbeam_search_decoder.py83 shape_t = array_ops.shape(t)
94 ([shape_t[0] * multiplier], shape_t[1:]), 0))