Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/shim/
Dtest_util.h72 TfLiteIntArray* shape_arr = TfLiteIntArrayCreate(shape.size()); in ReallocDynamicTensor() local
/external/tensorflow/tensorflow/c/eager/
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
/external/pytorch/aten/src/ATen/native/quantized/cpu/
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
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc717 SmallVector<int32_t> shape_arr; in matchAndRewrite() local
Dlegalize_tfl.cc1706 SmallVector<int32_t> shape_arr; in matchAndRewrite() local