| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression_test.cc | 93 TEST_F(XlaExpressionTest, GetShape) { in TEST_F() argument
|
| D | xla_expression.cc | 198 xla::StatusOr<TensorShape> XlaExpression::GetShape() const { in GetShape() function in tensorflow::XlaExpression
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | xla_service.proto | 74 rpc GetShape(GetShapeRequest) returns (GetShapeResponse) { rpc
|
| D | grpc_service.cc | 127 ::grpc::Status GRPCService::GetShape(::grpc::ServerContext* context, in GetShape() function in xla::GRPCService
|
| D | grpc_stub.cc | 109 Status GRPCStub::GetShape(const GetShapeRequest* request, in GetShape() function in xla::GRPCStub
|
| /external/llvm-project/flang/include/flang/Evaluate/ |
| D | shape.h | 185 std::optional<Shape> GetShape(FoldingContext &context, const A &x) { in GetShape() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 226 const Shape& GetShape() const { return shape_; } in GetShape() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 34 inline Status GetShape(OpKernelContext* ctx, int input_index, in GetShape() function
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 56 std::vector<int> FlexModelTest::GetShape(int tensor_index) { in GetShape() function in tflite::flex::testing::FlexModelTest
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_types.cc | 27 llvm::Optional<llvm::ArrayRef<int64_t>> GetShape(mlir::Value value) { in GetShape() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 450 StatusOr<Shape> Client::GetShape(const GlobalData& data) { in GetShape() function in xla::Client
|
| D | xla_builder.cc | 228 StatusOr<Shape> XlaBuilder::GetShape(XlaOp op) const { in GetShape() function in xla::XlaBuilder
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 82 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape() function in tflite::__anona9a8652d0111::KernelUtilTest
|
| D | test_util.h | 470 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape() function
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 235 DenseElementsAttr GetShape(Value output_val) { in GetShape() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 1096 Status Service::GetShape(const GetShapeRequest* arg, GetShapeResponse* result) { in GetShape() function in xla::Service
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 1108 TensorShapeProto GetShape(const string& input_name) const { in GetShape() function in tensorflow::grappler::__anon5a8ea6d90111::Conv2DProcessor
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1424 PyObject* PyTapeTensor::GetShape() const { in GetShape() function in PyTapeTensor
|