Searched refs:num_input_bytes_accessed (Results 1 – 2 of 2) sorted by relevance
48 std::vector<int64> 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()
1633 node_costs->num_input_bytes_accessed = in PredictDefaultNodeCosts()1857 node_costs->num_input_bytes_accessed = {a_indices_input_size, in PredictSparseTensorDenseMatMul()1894 node_costs->num_input_bytes_accessed = {0}; in PredictIdentity()1913 node_costs->num_input_bytes_accessed = {0}; in PredictVariable()1939 node_costs->num_input_bytes_accessed = {0}; in PredictMetadata()1974 node_costs->num_input_bytes_accessed.reserve(op_info.inputs().size()); in PredictGatherOrSlice()1979 node_costs->num_input_bytes_accessed.push_back(input_size); in PredictGatherOrSlice()1993 node_costs->num_input_bytes_accessed.push_back( in PredictGatherOrSlice()2036 node_costs->num_input_bytes_accessed = {ref_input_size, indices_input_size, in PredictScatter()2193 node_costs->num_input_bytes_accessed = {input_size}; in PredictMaxPool()[all …]