Home
last modified time | relevance | path

Searched defs:flat_inputs (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dcompiler.cpp227 flat_inputs; in compileKernel() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_spec.py326 def _validate_inputs(self, flat_inputs): argument
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_lower_io_to_vector.c425 bool flat_inputs[MAX_SLOTS] = { 0 }; in nir_lower_io_to_vector_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc392 std::vector<tensorflow::Tensor> flat_inputs; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_v2.py1063 def _raise_error_for_inputs_not_on_cpu(self, flat_inputs, flat_paths): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc749 flat_inputs; in RunMultipleSignatures() local
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2054 Slice<mpy::handle> flat_inputs; member
2401 Slice<mpy::handle> flat_inputs; in getsetitem_flat() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_compiler.h718 uint32_t flat_inputs; member
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_compiler.h885 uint32_t flat_inputs; member
/aosp_15_r20/external/mesa3d/src/gallium/drivers/iris/
H A Diris_context.h328 uint32_t flat_inputs; member