| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | parameterized_truncated_normal_op_test.cc | 29 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
|
| H A D | sparse_reduce_op.cc | 108 Status ValidateInputs(const Tensor *shape_t, const Tensor *reduction_axes_t) { in ValidateInputs()
163 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
275 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
|
| H A D | stateful_random_ops.cc | 132 const Tensor& shape_t = ctx->input(shape_input_idx); in StatefulRandomCompute() local
321 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| H A D | example_parsing_ops.cc | 1046 auto shape_t = sp_shape_d->vec<int64_t>(); in Compute() local
1062 auto shape_t = sp_shape_d->vec<int64_t>(); in Compute() local
1172 auto shape_t = sp_shape_d->vec<int64_t>(); in Compute() local
|
| H A D | sparse_softmax_op.cc | 44 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
|
| H A D | sparse_dense_binary_op_shared.cc | 62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
|
| H A D | random_binomial_op_test.cc | 30 Tensor shape_t(DT_INT32, TensorShape({2})); in RandomBinomialGraph() local
|
| H A D | stateless_random_gamma_op.cc | 202 const Tensor& shape_t = context->input(0); in Compute() local
|
| H A D | random_poisson_op.cc | 293 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| H A D | nccl_ops.cc | 233 const Tensor& shape_t = c->input(0); in ComputeAsync() local
|
| H A D | stateless_random_ops.cc | 76 const Tensor& shape_t = context->input(0); in Compute() local
|
| H A D | random_op.cc | 147 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| H A D | ctc_decoder_ops.cc | 151 auto shape_t = p_shape->vec<int64_t>(); in StoreAllDecodedSequences() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | random_crop_op.cc | 39 const Tensor& shape_t = context->input(1); in Compute() local
|
| H A D | image_ops.cc | 70 const Tensor& shape_t = ctx->input(2); in DoImageProjectiveTransformOp() local
|
| H A D | resize_nearest_neighbor_op.cc | 231 const Tensor& shape_t = context->input(1); in Compute() local
|
| H A D | resize_bicubic_op_test.cc | 264 auto shape_t = shape.flat<int32>(); in ResizeBicubic() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | image_resizer_state.h | 111 const Tensor& shape_t = context->input(1); in ValidateAndCalculateOutputSize() local
|
| H A D | example_proto_helper.cc | 391 auto shape_t = (*output_sparse_shapes_tensor)[d].vec<int64_t>(); in BatchExampleProtoToTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | nn_ops_test.cc | 571 Tensor shape_t = test::AsTensor<int32>(shape); in TEST() local
|
| H A D | array_ops_test.cc | 568 Tensor shape_t(DT_INT64, TensorShape{3}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 1879 Tensor shape_t; in ReplaceOperationWithBroadcastTo() local
|