| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | cudnn_rnn_ops.cc | 249 auto params_shape = c->input(3); in __anonb0821bee0602() local
286 auto params_shape = c->input(3); in __anonb0821bee0702() local
326 auto params_shape = c->input(3); in __anonb0821bee0802() local
|
| H A D | array_ops.cc | 1205 ShapeHandle params_shape; in __anon38bbb0e81802() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 631 Output GetBatchIndices(const Scope& scope, const Output& params_shape, in GetBatchIndices()
659 Output BatchGatherGrad(const Scope& scope, Output params_shape, Output values, in BatchGatherGrad()
708 Output params_shape = Cast(colocate_scope, params_shape64, DT_INT32); in GatherV2Grad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scatter_nd_util.cc | 22 Status ValidateScatterNdUpdateShape(const TensorShape& params_shape, in ValidateScatterNdUpdateShape()
|
| H A D | scatter_nd_op.cc | 279 TensorShape params_shape; in DoCompute() local
777 Status PrepareAndValidateInputs(const TensorShape& params_shape, in PrepareAndValidateInputs()
|
| H A D | cudnn_rnn_ops.cc | 2046 const TensorShape& params_shape, in AllocateOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_grad.py | 595 def _GetBatchIndices(params_shape, indices, batch_dims): argument
617 def _BatchGatherGrad(params_shape, values, indices, batch_dims, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_gather_op_test.py | 393 params_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 572 inline GatherNdHelperResult GatherNdHelper(const RuntimeShape& params_shape, in GatherNdHelper()
603 inline TfLiteStatus GatherNd(const RuntimeShape& params_shape, in GatherNd()
631 inline TfLiteStatus GatherNdString(const RuntimeShape& params_shape, in GatherNdString()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | gather_op.cc | 279 TensorShape params_shape = context->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | gather_op_test.py | 593 def testBatchDimsMatchesPythonBatching(self, params_shape, indices_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| H A D | resource_variable_ops_test.py | 1589 def testGatherWithBatchDimsMatchesTensor(self, params_shape, indices_shape, argument
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 548 const int32_t* __restrict params_shape, in Gather()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 549 const int32_t* __restrict params_shape, in Gather()
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 548 const int32_t* __restrict params_shape, in Gather()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 415 std::vector<int64_t> params_shape; member
2919 std::vector<int64_t> params_shape = RandomDims(1); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 7237 std::vector<int> params_shape; in TEST_P() member
|