Searched defs:is_1d (Results 1 – 9 of 9) sorted by relevance
42 bool is_1d, OpKernelContext* context) { in OutputSparse()138 bool is_1d = TensorShapeUtils::IsVector(data.shape()); in Compute() local264 bool is_1d = shape.NumElements() == 1; in Compute() local329 bool is_1d = false; in Compute() local
399 bool is_1d = dense_shape.NumElements() == 1; in Compute() local
534 bool is_1d, in lp_build_depth_stencil_load_swizzled()672 bool is_1d, in lp_build_depth_stencil_write_swizzled()
2444 const bool is_1d = variant->key.resource_1d; in generate_unswizzled_blend() local
443 bool is_1d = false; in vir_image_emit_register_writes() local
520 bool is_1d = (1 == kSpatialDim); in _qconv_prepack_onednn() local
1431 bool is_1d = (1 == kSpatialDim); in _quantized_convolution_onednn() local
408 const int is_1d = tx_class != TX_CLASS_2D; in decode_coefs() local410 #define case_sz(sz, bin, ns, is_1d) \ in decode_coefs() argument
430 unsigned is_1d : 1; member