Searched defs:input_ragged_rank (Results 1 – 4 of 4) sorted by relevance
42 const int input_ragged_rank, const int output_ragged_rank, in BuildDecodeRaggedTensorGraph()93 int input_ragged_rank = 2; in TEST_F() local123 int input_ragged_rank = 2; in TEST_F() local172 int input_ragged_rank = 0; in TEST_F() local231 int input_ragged_rank = 1; in TEST_F() local302 int input_ragged_rank = -1; in TEST_F() local339 int input_ragged_rank = 1; in TEST_F() local374 int input_ragged_rank = 1; in TEST_F() local432 int input_ragged_rank = 1; in TEST_F() local459 int input_ragged_rank = -1; in TEST_F() local[all …]
33 const Tensor& encoded_variant, int input_ragged_rank, in RaggedComponentsFromVariant()129 const std::vector<int>& nested_dim_sizes, const int input_ragged_rank, in NestedStackRaggedTensors()
220 int64_t input_ragged_rank; in RaggedTensorFromVariantShapeFn() local
1958 input_ragged_rank=None, argument