Searched refs:num_elements_dim (Results 1 – 1 of 1) sorted by relevance
473 const shape_inference::DimensionHandle num_elements_dim = in RegisterConstantNode() local478 CHECK(context->ValueKnown(num_elements_dim)); in RegisterConstantNode()479 const int64 num_output_elements = context->Value(num_elements_dim); in RegisterConstantNode()927 const shape_inference::DimensionHandle num_elements_dim = in AppendNodeOutputParams() local929 const int64 num_output_elements = context->Value(num_elements_dim); in AppendNodeOutputParams()