| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | linalg_ops_common.h | 46 using TensorShapes = gtl::InlinedVector<TensorShape, 4>;
58 OpKernelContext* context, const TensorShapes& input_matrix_shapes) const { in ValidateInputMatrixShapes()
66 const TensorShapes& input_matrix_shapes);
69 OpKernelContext* context, const TensorShapes& input_matrix_shapes);
72 const TensorShapes& input_matrix_shapes);
76 const TensorShapes& input_matrix_shapes);
88 virtual TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
89 const TensorShapes& input_matrix_shapes) const { in GetOutputMatrixShapes()
99 virtual int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const { in GetCostPerUnit()
175 const TensorShapes& input_matrix_shapes,
[all …]
|
| D | svd_op_impl.h | 48 using TensorShapes = typename Base::TensorShapes; variable
52 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
56 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
57 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
62 return TensorShapes({TensorShape({min_size}), in GetOutputMatrixShapes()
66 return TensorShapes({TensorShape({min_size})}); in GetOutputMatrixShapes()
71 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | self_adjoint_eig_v2_op_impl.h | 43 using TensorShapes = typename Base::TensorShapes; variable
49 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
50 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
53 return TensorShapes({TensorShape({n}), TensorShape({n, n})}); in GetOutputMatrixShapes()
55 return TensorShapes({TensorShape({n})}); in GetOutputMatrixShapes()
|
| D | eig_op_impl.h | 43 using TensorShapes = typename Base::TensorShapes; variable
54 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
55 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
58 return TensorShapes({TensorShape({n}), TensorShape({n, n})}); in GetOutputMatrixShapes()
60 return TensorShapes({TensorShape({n})}); in GetOutputMatrixShapes()
|
| D | linalg_ops_common.cc | 34 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleMatrix()
45 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleSquareMatrix()
56 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSolver()
73 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSquareSolver()
92 TensorShapes input_matrix_shapes; in Compute()
97 TensorShapes output_matrix_shapes; in Compute()
119 TensorShapes* input_matrix_shapes, TensorShape* batch_shape) { in AnalyzeInputs()
162 OpKernelContext* context, const TensorShapes& input_matrix_shapes, in PrepareOutputs()
164 TensorShapes* output_matrix_shapes) { in PrepareOutputs()
222 const TensorShapes& input_matrix_shapes, const TensorOutputs& outputs, in ComputeTensorSlice()
[all …]
|
| D | self_adjoint_eig_op.cc | 38 using TensorShapes = typename Base::TensorShapes; typedef in tensorflow::SelfAdjointEigOp
44 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
45 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
47 return TensorShapes({TensorShape({d + 1, d})}); in GetOutputMatrixShapes()
|
| D | matrix_solve_ls_op_impl.h | 43 using TensorShapes = typename Base::TensorShapes; variable
55 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
59 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
60 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
61 return TensorShapes({TensorShape({input_matrix_shapes[0].dim_size(1), in GetOutputMatrixShapes()
65 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | cholesky_grad.cc | 32 using TensorShapes = typename Base::TensorShapes; typedef in tensorflow::CholeskyGrad
43 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
55 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
56 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
57 return TensorShapes({input_matrix_shapes[0]}); in GetOutputMatrixShapes()
|
| D | tridiagonal_matmul_op.cc | 38 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
62 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
63 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
64 return TensorShapes({input_matrix_shapes[3]}); in GetOutputMatrixShapes()
67 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | qr_op_impl.h | 62 using TensorShapes = typename Base::TensorShapes; variable
66 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
70 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
71 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
76 return TensorShapes({TensorShape({m, m}), TensorShape({m, n})}); in GetOutputMatrixShapes()
78 return TensorShapes( in GetOutputMatrixShapes()
83 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | tridiagonal_solve_op.cc | 50 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
72 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
73 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
74 return TensorShapes({input_matrix_shapes[1]}); in GetOutputMatrixShapes()
77 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | matrix_solve_op.cc | 55 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
59 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
60 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
61 return TensorShapes({TensorShape({input_matrix_shapes[0].dim_size(1), in GetOutputMatrixShapes()
65 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| D | determinant_op.cc | 84 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
85 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
86 return TensorShapes({TensorShape({}), TensorShape({})}); in GetOutputMatrixShapes()
108 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
109 const TensorShapes& input_matrix_shape) const final { in GetOutputMatrixShapes()
110 return TensorShapes({TensorShape({})}); in GetOutputMatrixShapes()
|
| D | lu_op.cc | 35 using TensorShapes = gtl::InlinedVector<TensorShape, 4>; typedef in tensorflow::LuOp
96 TensorShapes output_matrix_shapes({input.shape(), permutation_shape}); in Compute()
131 const TensorShapes& output_matrix_shapes) { in ComputeTensorSlice()
|
| D | tridiagonal_solve_op_gpu.cu.cc | 107 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
130 TensorShapes GetOutputMatrixShapes( in GetOutputMatrixShapes()
131 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
132 return TensorShapes({input_matrix_shapes[1]}); in GetOutputMatrixShapes()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op_test.cc | 29 : DatasetParams(TensorDtypes(components), TensorShapes(components), in TensorDatasetParams()
61 std::vector<PartialTensorShape> TensorShapes( in TensorShapes() function in tensorflow::data::__anond871d70a0111::TensorDatasetParams
|