Searched refs:CreateShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | custom_call.md | 33 xla::Parameter(0, xla::ShapeUtil::CreateShape(F32, {128}), "p0");
35 xla::Parameter(1, xla::ShapeUtil::CreateShape(F32, {2048}), "p1");
38 /*output_shape=*/ShapeUtil::CreateShape(F32, {2048}));
112 /*output_shape=*/ShapeUtil::CreateShape(F32, {2048}),
|
| /external/llvm-project/flang/lib/Evaluate/ |
| D | shape.cpp | 401 return Result{CreateShape(n, base)}; in operator ()()
420 return Result{CreateShape(n, base)}; in operator ()()
452 return CreateShape(rank, base); in operator ()()
455 return Result{CreateShape(rank, base)}; in operator ()()
|
| /external/llvm-project/flang/include/flang/Evaluate/ |
| D | shape.h | 129 Shape CreateShape(int rank, NamedEntity &base) const { in CreateShape() function
|