Searched refs:num_input_bytes_accessed (Results 1 – 2 of 2) sorted by relevance
1665 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 …]
48 std::vector<int64_t> num_input_bytes_accessed; // ordered by input tensors. member55 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()