Searched defs:slice_inputs (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | increase_dynamism_for_auto_jit_pass.cc | 103 SliceInputs slice_inputs; in GetSliceInputs() local
135 const SliceInputs& slice_inputs) { in MakeSliceIndexAndSizeInt64()
173 const SliceInputs& slice_inputs, in ComputeSliceSize()
237 Graph* g, Node* slice, const SliceInputs& slice_inputs, in ConvertTensorFlowSliceToStaticShapedSlice()
305 Status RewriteSlice(Graph* g, Node* slice, const SliceInputs& slice_inputs, in RewriteSlice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | data_adapter.py | 344 def slice_inputs(self, indices_dataset, inputs): member in TensorLikeDataAdapter
449 def slice_inputs(self, indices_dataset, inputs): member in GenericArrayLikeDataAdapter
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_legacy_vmap.py | 848 def slice_inputs(inputs, bdims, i): function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 1265 def slice_inputs(inputs, bdims, i): function
|