Home
last modified time | relevance | path

Searched defs:batch_axis (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dreverse_sequence_op_test.py35 batch_axis, argument
57 batch_axis, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dreverse_sequence_op_test.py29 batch_axis, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Drnn.py456 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
H A Darray_ops.py4982 batch_axis=None, argument
5043 batch_axis=None, argument
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DTensor.cpp184 const int64_t batch_axis = axis_map.at(concatted_whcn_dim); in calculate_image_extents() local