Home
last modified time | relevance | path

Searched defs:shape_t (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dparameterized_truncated_normal_op_test.cc29 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormal() local
58 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormal2SD() local
85 Tensor shape_t(DT_INT32, TensorShape({2})); in PTruncatedNormalOneTail() local
Dsparse_reduce_op.cc106 Status ValidateInputs(const Tensor *shape_t, const Tensor *reduction_axes_t) { in ValidateInputs()
161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
257 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
Dstateful_random_ops.cc105 const Tensor& shape_t = ctx->input(shape_input_idx); in ComputeImpl() local
154 const Tensor& shape_t = ctx->input(0); in Compute() local
Drandom_crop_op.cc39 const Tensor& shape_t = context->input(1); in Compute() local
Dsparse_dense_binary_op_shared.cc62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
Dsparse_softmax_op.cc43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
Dstateless_random_ops.cc74 const Tensor& shape_t = context->input(0); in Compute() local
Dexample_parsing_ops.cc690 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
706 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
816 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
Dimage_resizer_state.h87 const Tensor& shape_t = context->input(1); in ValidateAndCalculateOutputSize() local
Drandom_poisson_op.cc289 const Tensor& shape_t = ctx->input(0); in Compute() local
Dresize_nearest_neighbor_op.cc191 const Tensor& shape_t = context->input(1); in Compute() local
Dnccl_ops.cc233 const Tensor& shape_t = c->input(0); in ComputeAsync() local
Dresize_bicubic_op_test.cc260 auto shape_t = shape.flat<int32>(); in ResizeBicubic() local
Dctc_decoder_ops.cc146 auto shape_t = p_shape->vec<int64>(); in StoreAllDecodedSequences() local
Drandom_op.cc275 const Tensor& shape_t = ctx->input(0); in Compute() local
/external/tensorflow/tensorflow/contrib/image/kernels/
Dimage_ops.cc88 const Tensor& shape_t = ctx->input(2); in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops_test.cc536 Tensor shape_t = test::AsTensor<int32>(shape); in TEST() local
Darray_ops_test.cc527 Tensor shape_t(DT_INT64, TensorShape{3}); in TEST() local
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc388 auto shape_t = (*output_sparse_shapes_tensor)[d].vec<int64>(); in BatchExampleProtoToTensors() local