Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for add_int_val (0.13 sec)

  1. tensorflow/c/kernels_test.cc

          tensor_proto.mutable_tensor_shape()->add_dim()->set_size(3);
          tensor_proto.set_dtype(DT_INT32);
          tensor_proto.add_int_val(1);
          tensor_proto.add_int_val(2);
          tensor_proto.add_int_val(3);
          tensor_proto.add_int_val(4);
          tensor_proto.add_int_val(5);
          tensor_proto.add_int_val(6);
          return tensor_proto;
        }
      };
    
      auto my_create_func = [](TF_OpKernelConstruction* ctx) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/shape_inference.cc

                  shape_proto.dim_size());
              for (const auto& dim : shape_proto.dim()) {
                if (dtype == DT_INT32) {
                  value.add_int_val(dim.size());
                } else {
                  value.add_int64_val(dim.size());
                }
              }
              AddNodeAttr("value", value, &const_def);
              for (auto const& attr : n->attrs()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

        tensorflow::TensorShapeProto shape;
        shape.add_dim()->set_size(2);
        *tensor_proto.mutable_tensor_shape() = shape;
        for (int i = 0; i < 2; ++i) {
          tensor_proto.add_int_val(1);
        }
        NodeDef const_def;
        TF_CHECK_OK(NodeDefBuilder("const", "Const")
                        .Attr("dtype", DT_INT32)
                        .Attr("value", tensor_proto)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
Back to top