Searched defs:index_tensor (Results 1 – 6 of 6) sorted by relevance
122 auto index_tensor = graph->insert( in MergeSliceAndSelectToIndices() local141 auto index_tensor = ConvertSliceToIndex(node, size, index_put_node); in MergeSliceAndSelectToIndices() local147 auto index_tensor = ConvertSelectToIndex(node->input(2), index_put_node); in MergeSliceAndSelectToIndices() local
167 const Tensor& index_tensor = context->input(0); in Compute() local
347 Tensor index_tensor(DT_INT64, TensorShape({})); in InsertOneLocked() local
168 static inline void recordSelectTrace(const Tensor& index_tensor) { in recordSelectTrace()
563 def index_tensor(fake_mode, func, *args, **kwargs): function
6388 auto index_tensor = in matchAndRewrite() local