Searched refs:GraphTensor (Results 1 – 1 of 1) sorted by relevance
47 class GraphTensor;54 class GraphTensor : public TracingTensorHandle { class56 explicit GraphTensor(TF_Output output, TF_Graph* graph) in GraphTensor() function in tensorflow::tracing::graph::GraphTensor136 GraphTensor* t = dyn_cast<GraphTensor>(input); in AddInput()147 GraphTensor* t = dyn_cast<GraphTensor>(inputs[i]); in AddInputList()169 retvals[i] = new GraphTensor({operation, i}, g_); in Execute()350 auto* t = dyn_cast<GraphTensor>(outputs[0]); in AddParameter()364 GraphTensor* output = dyn_cast<GraphTensor>(abstract_output); in Finalize()