Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for UnknownShape (0.19 sec)

  1. tensorflow/c/c_api_experimental.cc

        std::vector<DimensionHandle> dims;
        const TF_ShapeAndType& input_shape = input_shapes->items[i];
        if (input_shape.num_dims == InferenceContext::kUnknownRank) {
          c.SetInput(i, c.UnknownShape());
          continue;
        }
        dims.reserve(input_shape.num_dims);
        for (int j = 0; j < input_shape.num_dims; ++j) {
          dims.push_back(c.MakeDim(input_shape.dims[j]));
        }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

          .Attr("v: " #type)                                      \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape); \
      REGISTER_OP("CApiAttributesTestOpList" #type)               \
          .Attr("v: list(" #type ")")                             \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape)
    ATTR_TEST_REGISTER_OP(string);
    ATTR_TEST_REGISTER_OP(int);
    ATTR_TEST_REGISTER_OP(float);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  3. tensorflow/c/c_api_function_test.cc

      ASSERT_NE(*func, nullptr);
    }
    
    REGISTER_OP("CustomOp")
        .Output("output: float32")
        .Attr("index: int")
        .SetShapeFn(tensorflow::shape_inference::UnknownShape);
    
    void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
                                       const char* name, const char* placeholder,
                                       TF_Operation** op) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  4. tensorflow/c/c_api.cc

        dim_vec.reserve(num_dims);
        for (int i = 0; i < num_dims; ++i) {
          dim_vec.push_back(ic->MakeDim(dims[i]));
        }
        return ic->MakeShape(dim_vec);
      } else {
        return ic->UnknownShape();
      }
    }
    
    }  // namespace
    
    void TF_GraphSetOutputHandleShapesAndTypes(TF_Graph* graph, TF_Output output,
                                               int num_shapes_and_types,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top