Home
last modified time | relevance | path

Searched refs:num_input_bytes_accessed (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc1665 node_costs->num_input_bytes_accessed = in PredictDefaultNodeCosts()
1891 node_costs->num_input_bytes_accessed = {a_indices_input_size, in PredictSparseTensorDenseMatMul()
1928 node_costs->num_input_bytes_accessed = {0}; in PredictIdentity()
1947 node_costs->num_input_bytes_accessed = {0}; in PredictVariable()
1973 node_costs->num_input_bytes_accessed = {0}; in PredictMetadata()
2008 node_costs->num_input_bytes_accessed.reserve(op_info.inputs().size()); in PredictGatherOrSlice()
2013 node_costs->num_input_bytes_accessed.push_back(input_size); in PredictGatherOrSlice()
2027 node_costs->num_input_bytes_accessed.push_back( in PredictGatherOrSlice()
2070 node_costs->num_input_bytes_accessed = {ref_input_size, indices_input_size, in PredictScatter()
2233 node_costs->num_input_bytes_accessed = {input_size}; in PredictMaxPool()
[all …]
H A Dop_level_cost_estimator.h48 std::vector<int64_t> num_input_bytes_accessed; // ordered by input tensors. member
55 return std::accumulate(num_input_bytes_accessed.begin(), in num_total_input_bytes()
56 num_input_bytes_accessed.end(), 0LL); in num_total_input_bytes()