| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_const_op_test.py | 178 inner_shape=None, argument
296 inner_shape=None, argument
385 inner_shape=None, argument
|
| H A D | ragged_factory_ops.py | 35 def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
88 def constant_value(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
147 inner_shape): argument
|
| H A D | dynamic_ragged_shape_test.py | 217 def testFromTensor(self, value, row_partitions, inner_shape): argument
291 inner_shape, argument
1042 inner_shape, argument
2022 inner_shape, argument
2262 def foo(inner_shape): argument
2298 def foo(inner_shape): argument
2311 def foo(inner_shape): argument
2915 inner_shape, argument
3615 def test_constructor_idempotent(self, row_partitions, static_inner_shape, argument
|
| H A D | ragged_constant_value_op_test.py | 187 inner_shape=None, argument
306 inner_shape=None, argument
|
| H A D | ragged_to_tensor_op_test.py | 323 inner_shape=None, argument
|
| H A D | dynamic_ragged_shape.py | 424 def _from_inner_shape(cls, inner_shape, dtype=None): argument
730 def inner_shape(self): member in DynamicRaggedShape
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/ |
| H A D | reconstruction_ops.py | 77 def full_shape(inner_shape): argument
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | GatherNdLayer.cpp | 61 std::vector<unsigned int> inner_shape; in InferOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 668 auto inner_shape = in BatchGatherGrad() local
728 auto inner_shape = in GatherV2Grad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | dot_op_emitter.cc | 1309 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray() local
|