Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for proto2 (0.28 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.cc

        }
        if (num_dims_i < 0) {
          proto[i].set_unknown_rank(true);
        } else {
          const int64_t* dims_i = dims[i];
          auto proto_i = &proto[i];
          for (int d = 0; d < num_dims_i; ++d) {
            proto_i->add_dim()->set_size(dims_i[d]);
          }
        }
      }
      forward_op_.attrs.Set(
          attr_name, gtl::ArraySlice<TensorShapeProto>(proto.get(), num_values));
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  2. tensorflow/c/checkpoint_reader.cc

            << entry.InitializationErrorString();
        for (int i = 0; i < entry.slices_size(); ++i) {
          const auto& slice_proto = entry.slices(i);
          CHECK(filtered_keys
                    .insert(EncodeTensorNameSlice(
                        string(v2_reader_->key()) /* full var's name */,
                        TensorSlice(slice_proto)))
                    .second);
        }
      }
    
      // Second pass: adds the entries, ignoring the filtered keys.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
Back to top