| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| D | sparse_reduce_op.cc | 106 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
|
| D | sparse_softmax_op.cc | 43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
|
| D | sparse_dense_binary_op_shared.cc | 62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
|
| D | random_binomial_op_test.cc | 30 Tensor shape_t(DT_INT32, TensorShape({2})); in RandomBinomialGraph() local
|
| 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
|
| D | stateless_random_gamma_op.cc | 202 const Tensor& shape_t = context->input(0); in Compute() local
|
| D | random_poisson_op.cc | 293 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| D | example_parsing_ops.cc | 1046 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
1062 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
1172 auto shape_t = sp_shape_d->vec<int64>(); in Compute() local
|
| D | stateless_random_ops.cc | 76 const Tensor& shape_t = context->input(0); in Compute() local
|
| D | nccl_ops.cc | 233 const Tensor& shape_t = c->input(0); in ComputeAsync() local
|
| D | random_op.cc | 147 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| D | stateless_random_ops_v2.cc | 73 const Tensor& shape_t = ctx->input(0); in Compute() local
|
| D | ctc_decoder_ops.cc | 148 auto shape_t = p_shape->vec<int64>(); in StoreAllDecodedSequences() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | random_crop_op.cc | 39 const Tensor& shape_t = context->input(1); in Compute() local
|
| D | image_ops.cc | 70 const Tensor& shape_t = ctx->input(2); in DoImageProjectiveTransformOp() local
|
| D | resize_nearest_neighbor_op.cc | 218 const Tensor& shape_t = context->input(1); in Compute() local
|
| D | resize_bicubic_op_test.cc | 262 auto shape_t = shape.flat<int32>(); in ResizeBicubic() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | image_resizer_state.h | 89 const Tensor& shape_t = context->input(1); in ValidateAndCalculateOutputSize() local
|
| D | example_proto_helper.cc | 391 auto shape_t = (*output_sparse_shapes_tensor)[d].vec<int64>(); in BatchExampleProtoToTensors() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops_test.cc | 571 Tensor shape_t = test::AsTensor<int32>(shape); in TEST() local
|
| D | array_ops_test.cc | 558 Tensor shape_t(DT_INT64, TensorShape{3}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1830 Tensor shape_t; in ReplaceOperationWithBroadcastTo() local
|