Home
last modified time | relevance | path

Searched defs:TensorShapes (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dself_adjoint_eig_v2_op_impl.h43 using TensorShapes = typename Base::TensorShapes; variable
H A Dself_adjoint_eig_op.cc38 using TensorShapes = typename Base::TensorShapes; typedef in tensorflow::SelfAdjointEigOp
H A Deig_op_impl.h43 using TensorShapes = typename Base::TensorShapes; variable
H A Dsvd_op_impl.h48 using TensorShapes = typename Base::TensorShapes; variable
H A Dmatrix_solve_ls_op_impl.h43 using TensorShapes = typename Base::TensorShapes; variable
H A Dlu_op.cc35 using TensorShapes = gtl::InlinedVector<TensorShape, 4>; typedef in tensorflow::LuOp
H A Dcholesky_grad.cc32 using TensorShapes = typename Base::TensorShapes; typedef in tensorflow::CholeskyGrad
H A Dqr_op_impl.h62 using TensorShapes = typename Base::TensorShapes; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dtensor_dataset_op_test.cc66 std::vector<PartialTensorShape> TensorShapes( in TensorShapes() function in tensorflow::data::__anone67d5dc00111::TensorDatasetParams
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_irparser.cpp314 TEST(IRParserTest, TensorShapes) { argument