HomeSort by relevance Sort by last modified time
    Searched refs:input_tensor_id (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/c/eager/
tape.h 44 std::vector<int64> input_tensor_id; member in struct:tensorflow::eager::OpTapeEntry
139 gtl::ArraySlice<int64> input_tensor_id,
190 gtl::ArraySlice<int64> input_tensor_id, BackwardFunction* backward_function,
192 if (!ShouldRecord(input_tensor_id)) {
197 ids.reserve(input_tensor_id.size());
198 for (int64 i : input_tensor_id) {
245 for (int64 id : op_it->second.input_tensor_id) {
327 for (auto it : op_it->second.input_tensor_id) {
528 in_gradients.resize(trace.input_tensor_id.size());
539 << trace.input_tensor_id.size() << " sources"
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
utils.cc 99 TensorId input_tensor_id = ParseTensorName(input_name); local
100 const string input_node_name = input_tensor_id.first.ToString();
172 TensorId input_tensor_id = ParseTensorName(input_name); local
173 const string input_node_name = input_tensor_id.first.ToString();
174 const int output_index = input_tensor_id.second;

Completed in 954 milliseconds