Home
last modified time | relevance | path

Searched defs:shape_arr (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Ddlpack.cc249 bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, in IsValidStrideCompactRowMajorData()
302 std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; in TFE_HandleToDLPack() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtest_util.h72 TfLiteIntArray* shape_arr = TfLiteIntArrayCreate(shape.size()); in ReallocDynamicTensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag.cpp281 std::array<int64_t, 3> shape_arr; in embedding_bag_nbit_impl() local
451 std::array<int64_t, 3> shape_arr; in embedding_bag_byte_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc717 SmallVector<int32_t> shape_arr; in matchAndRewrite() local
H A Dlegalize_tfl.cc1706 SmallVector<int32_t> shape_arr; in matchAndRewrite() local