Searched refs:lookup_data (Results 1 – 5 of 5) sorted by relevance
78 const int32_t* lookup_data = GetTensorData<int32_t>(lookup); in EvalSimple() local80 int idx = lookup_data[i]; in EvalSimple()110 const int32_t* lookup_data = GetTensorData<int32_t>(lookup); in EvalHybrid() local113 int idx = lookup_data[i]; in EvalHybrid()
56 void SetInput(std::initializer_list<int> lookup_data, in SetInput() argument60 PopulateTensor(lookup_, lookup_data); in SetInput()
559 lookup_data = self._index_lookup_layer(inputs)566 if tf_utils.is_ragged(lookup_data):567 dense_data = lookup_data.to_tensor(default_value=0)569 dense_data = lookup_data586 return lookup_data
56471 …<var-decl name="lookup_data" type-id="eaa32e2f" visibility="default" filepath="include/net/fib_rul…
56456 …<var-decl name="lookup_data" type-id="eaa32e2f" visibility="default" filepath="include/net/fib_rul…