Searched refs:TracingTensorHandle (Results 1 – 8 of 8) sorted by relevance
57 class TracingTensorHandle : public AbstractTensorHandle {59 explicit TracingTensorHandle(AbstractTensorHandleKind kind) in TracingTensorHandle() function112 TracingTensorHandle**) = 0;
53 class GraphTensor : public TracingTensorHandle {56 : TracingTensorHandle(kGraph), output_(output), graph_(graph) {} in GraphTensor()360 TracingTensorHandle** output) override { in AddParameter()385 *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]); in AddParameter()
147 tracing::TracingTensorHandle* x_raw = nullptr; in TEST_P()171 tracing::TracingTensorHandle* x_raw = nullptr; in TEST_P()
109 using tensorflow::tracing::TracingTensorHandle;140 TracingTensorHandle* t; in TF_AddFunctionParameter()
40 tracing::TracingTensorHandle* handle = nullptr; in CreateParamsForInputs()
226 tracing::TracingTensorHandle* handle = nullptr; in CreateParamsForInputs()
56 using tensorflow::tracing::TracingTensorHandle;135 TracingTensorHandle* handle = nullptr; in PYBIND11_MODULE()161 if (!isa<TracingTensorHandle>(handle)) { in PYBIND11_MODULE()
75 using tensorflow::tracing::TracingTensorHandle;93 class MlirTensor : public TracingTensorHandle {96 : TracingTensorHandle(kMlir), value_(value) {} in MlirTensor()263 TracingTensorHandle** handle) override;562 TracingTensorHandle** handle) { in AddParameter()