Searched defs:input_elements (Results 1 – 6 of 6) sorted by relevance
76 const std::vector<std::vector<Tensor>>& input_elements) { in ListInputTypes()87 const std::vector<std::vector<Tensor>>& input_elements) { in ListOutputTypes()96 const std::vector<std::vector<Tensor>>& input_elements) { in ListOutputShapes()
693 const size_t input_elements = input_height * input_width; in setup_average_pooling2d() local
1086 DatasetInput input_elements = GetDatasetInput(op->getOperand(0)); in InferShapeForDatasetOpCommon() local1164 DatasetInput input_elements = GetDatasetInput(op.input_dataset()); in InferShapeForReduceDataset() local
713 size_t input_elements; member740 size_t input_elements; member
925 const int input_elements = input_shape.FlatSize(); in VerifySparsity() local
646 size_t input_elements; member