Home
last modified time | relevance | path

Searched defs:tensor_id (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view.h159 TensorId tensor_id = ParseTensorName(port.node->input(i)); in GetFanin() local
177 TensorId tensor_id = ParseTensorName(port.node->input(port.port_id)); in GetRegularFanin() local
236 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFanins() local
307 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFaninEdges() local
340 TensorId tensor_id = ParseTensorName(node->input(i)); in AddFanouts() local
Dmutable_graph_view.cc43 bool IsTensorIdPortValid(const TensorId& tensor_id) { in IsTensorIdPortValid()
47 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular()
51 bool IsTensorIdControlling(const TensorId& tensor_id) { in IsTensorIdControlling()
65 TensorId tensor_id = ParseTensorName(node.input(0)); in IsIdentityConsumingSwitch() local
122 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs() local
348 TensorId tensor_id = ParseTensorName(node->input(pos)); in AddAndDedupFanouts() local
655 TensorId tensor_id = ParseTensorName(node->input(i)); in SwapNodeNames() local
933 TensorId tensor_id = ParseTensorName(node->input(i)); in AddRegularFaninByPort() local
961 TensorId tensor_id(fanin.node->name(), fanin.port_id); in GetControllingFaninToAdd() local
1064 TensorId tensor_id = ParseTensorName(node->input(i)); in RemoveRegularFaninInternal() local
[all …]
Dutils.cc160 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString()
176 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput()
Dmutable_graph_view_test.cc40 TensorId tensor_id = ParseTensorName(fanins[i]); in CompareNodeFanins() local
62 TensorId tensor_id = ParseTensorName(fanout); in CompareNodeFanouts() local
1467 TensorId tensor_id; in TestRemoveAllFanins() local
/external/tensorflow/tensorflow/core/graph/
Dtensor_id.cc65 bool IsTensorIdControl(const TensorId& tensor_id) { in IsTensorIdControl()
Dtensor_id_test.cc95 TensorId tensor_id = ParseTensorName(input); in TEST() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc8443 TF_Tensor* TF_DequeueNamedTensor(TF_Session* session, int tensor_id, in TF_DequeueNamedTensor()
8482 void TF_EnqueueNamedTensor(TF_Session* session, int tensor_id, in TF_EnqueueNamedTensor()
8571 int tensor_id, TF_Status* status) { in createTFEQueue()
8643 TFE_TensorHandle* TFE_DequeueNamedTensor(TF_Session* session, int tensor_id, in TFE_DequeueNamedTensor()
8663 TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_DequeueNamedTensorFromCtx()
8676 void TFE_EnqueueNamedTensor(TF_Session* session, int tensor_id, in TFE_EnqueueNamedTensor()
8695 void TFE_EnqueueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_EnqueueNamedTensorFromCtx()
8709 void TFE_EnqueueVariantTensor(TF_Session* session, int tensor_id, in TFE_EnqueueVariantTensor()
8726 TFE_TensorHandle* TFE_DequeueVariantTensor(TF_Session* session, int tensor_id, in TFE_DequeueVariantTensor()
/external/tensorflow/tensorflow/python/eager/
Dtape.py138 def delete_trace(tensor_id): argument
Dpywrap_tfe_src.cc1441 tensorflow::int64 tensor_id = FastTensorId(tensor); local
1650 void TFE_Py_TapeSetDeleteTrace(tensorflow::int64 tensor_id) {
/external/tensorflow/tensorflow/c/eager/
Dtape.h201 int64 tensor_id) { in Watch()
238 int64 tensor_id) { in DeleteTrace()
334 int64 tensor_id = tensor_stack.back(); in PrepareBackprop() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.cc52 SafeTensorId tensor_id = ParseTensorName(tensor); in GetTensorProperties() local
Dfunction_optimizer.cc1629 SafeTensorId tensor_id = ParseTensorName(input); in InlineIndirectFunctionCall() local
1670 SafeTensorId tensor_id = ParseTensorName(output_tensor); in InlineIndirectFunctionCall() local
Darithmetic_optimizer.cc1559 TensorId tensor_id = ParseTensorName(output->input(0)); in InitializeChains() local
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc272 explicit SymbolPredicate(int64 id, TensorId tensor_id, bool must_be_true) in SymbolPredicate()
290 TensorId tensor_id() const { return tensor_id_; } in tensor_id() function in tensorflow::__anon9db8ce5c0111::SymbolPredicate
370 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() local
1162 for (TensorId tensor_id : tensor_ids) { in Print() local
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate.cc268 &model_state_inputs, &model_state_outputs](int tensor_id) { in AddOpsAndParams()
716 auto tensor_id = buf[j]; in BuildGraph() local
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc299 &model_state_inputs, &model_state_outputs](int tensor_id) { in AddOpsAndParams()
727 auto tensor_id = buf[j]; in BuildGraph() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1196 int tensor_id = node->inputs->data[input_tensor_index]; in AddDequantizeOperatorsWhereNeeded() local
/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm_test.cc211 void PopulateWeightTensor(int tensor_id, const std::vector<float>& f) { in PopulateWeightTensor()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2045 SafeTensorId tensor_id = ParseTensorName(feed.first); in InferStatically() local
/external/tensorflow/tensorflow/python/framework/
Dops.py74 def tensor_id(tensor): function