Home
last modified time | relevance | path

Searched defs:func_inputs (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cpu_cpp_wrapper.py123 func_inputs=None, argument
178 func_inputs: list = None variable in BaseTest
H A Dtest_cuda_cpp_wrapper.py92 func_inputs=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc509 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local
574 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local
742 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local
814 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local
1030 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dfunctional_grad.cc36 std::vector<Output> func_inputs; in PartitionedCallGrad() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dunified_api_testutil.cc65 std::vector<AbstractTensorHandle*> func_inputs; in RunModel() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc768 std::vector<tensorflow::NodeDefBuilder::NodeOut> func_inputs; in PrepareGraphForMlir() local
1110 std::vector<NodeDefBuilder::NodeOut> func_inputs; in InsertFunctionForTPUEmbeddingCheckpoint() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dfunction_cache_test.cc128 std::vector<tensorflow::AbstractTensorHandle*> func_inputs; in PrepareFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Ddecompose_reduce_dataset.cc288 auto func_inputs = reduce_func.getFunctionType().getInputs(); in DecomposeReduceDatasetInFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Draise_target_subgraphs.cc287 SmallVector<Value, 4> func_inputs; in ExtractSubgraphToFunc() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc1019 std::vector<tensorflow::NodeDefBuilder::NodeOut> func_inputs; in MakeOuterGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc1182 std::vector<int> func_inputs = subgraph.inputs; in ConvertSubgraph() local