| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.h | 38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {}
62 const Shape& shape() const { return shape_; }
77 string ToString() const { return ShapeUtil::HumanStringWithLayout(shape_); }
84 Shape shape_; member in class:xla::ShapeLayout
|
| shape_tree.h | 126 CHECK_EQ(other.shape_, other.shape_storage_.get());
127 shape_storage_.reset(new Shape(*other.shape_));
128 shape_ = shape_storage_.get();
130 shape_ = other.shape_;
144 const Shape& shape() const { return *shape_; }
155 shape_ = shape;
312 // If we own our Shape, this field contains it, and shape_ is a pointer into
318 const Shape* shape_; member in class:xla::ShapeTree
496 shape_(shape_storage_.get())
[all...] |
| literal_util.h | 90 const Shape& shape() const { return shape_; }
94 Shape* mutable_shape_do_not_use() { return &shape_; }
695 // subshape within the shape in the containing Literal (Literal::shape_).
761 Shape shape_; member in class:xla::Literal
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| shape_partition.h | 67 ShapePartitionAssigner(const Shape& shape) : shape_(shape) {}
77 const Shape& shape_; member in class:xla::cpu::ShapePartitionAssigner
94 const Shape& shape_; member in class:xla::cpu::ShapePartitionIterator
|
| xfeed_manager_test.cc | 36 : shape_(ShapeUtil::MakeShape(U8, {length})),
48 EXPECT_EQ(expect_shape_match_, ShapeUtil::Equal(shape_, shape.ValueOrDie()))
49 << "want " << ShapeUtil::HumanString(shape_) << " "
54 const Shape& shape() const { return shape_; }
57 Shape shape_; member in class:xla::__anon38890::TestInfeedBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| loop_emitter.h | 81 Shape shape_; member in class:xla::llvm_ir::LoopEmitter
|
| ir_array.h | 168 IrArray() : base_ptr_(nullptr), shape_(nullptr) {}
184 CHECK(shape_ != nullptr);
185 return *shape_;
282 const Shape* shape_; member in class:xla::llvm_ir::IrArray
|
| /external/tensorflow/tensorflow/core/kernels/ |
| immutable_constant_op.h | 43 TensorShape shape_; member in class:tensorflow::ImmutableConstantOp
|
| variable_ops.h | 62 TensorShape shape_; member in class:tensorflow::VariableOp
|
| conditional_accumulator_base_op.h | 52 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
82 PartialTensorShape shape_; member in class:tensorflow::ConditionalAccumulatorBaseOp
|
| inplace_ops.cc | 125 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &shape_));
134 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, shape_, &out, attr));
138 TensorShape shape_; member in class:tensorflow::__anon39816::ParallelConcatStart
|
| conditional_accumulator_base.h | 126 const PartialTensorShape shape_; member in class:tensorflow::ConditionalAccumulatorBase
|
| variable_ops.cc | 51 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
64 (*var)->tensor()->set_shape(shape_);
88 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
103 s = context->allocate_temp(dtype_, shape_, &tmp_var->val);
126 TensorShape shape_; member in class:tensorflow::TemporaryVariableOp
|
| /external/libtextclassifier/ |
| tensor-view.h | 36 : data_(data), shape_(shape), size_(internal::NumberOfElements(shape)) {}
46 const std::vector<int>& shape() const { return shape_; }
48 int dim(int i) const { return shape_[i]; }
50 int dims() const { return shape_.size(); }
66 const std::vector<int> shape_; member in class:libtextclassifier2::TensorView
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| sendrecv_ops.cc | 60 xla::Shape shape_; member in class:tensorflow::__anon38774::RecvOp
72 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype, tensor_shape, &shape_));
79 ctx->SetOutput(0, ctx->builder()->Recv(shape_, channel));
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_slice_set.h | 45 const TensorShape& shape() const { return shape_; }
85 const TensorShape shape_; member in class:tensorflow::checkpoint::TensorSliceSet
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_resource.h | 70 const TensorShape& shape() const { return shape_; }
146 TensorShape shape_; member in class:tensorflow::XlaResource
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| defuser_test.cc | 42 const Shape shape_ = ShapeUtil::MakeShape(F32, {2, 2}); member in class:xla::__anon38892::DefuserTest
48 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
50 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
52 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
63 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
65 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
67 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
86 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
88 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
90 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1))
[all...] |
| hlo_instruction.h | 519 Shape* mutable_shape() { return &shape_; }
1370 Shape shape_; member in class:xla::HloInstruction
[all...] |
| /external/webrtc/webrtc/modules/desktop_capture/ |
| desktop_frame.h | 59 const DesktopRegion* shape() const { return shape_.get(); }
60 void set_shape(DesktopRegion* shape) { shape_.reset(shape); }
91 rtc::scoped_ptr<DesktopRegion> shape_; member in class:webrtc::DesktopFrame
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor.h | 122 DataType dtype() const { return shape_.data_type(); }
125 const TensorShape& shape() const { return shape_; }
467 void set_dtype(DataType t) { shape_.set_data_type(t); }
475 TensorShape shape_; member in class:tensorflow::Tensor
509 shape_ = shape;
599 CHECK_EQ(sizeof(T), shape_.dim_sizes()[NDIMS] * DataTypeSize(dtype()));
602 dims[d] = shape_.dim_sizes()[d];
615 CHECK_EQ(sizeof(T), shape_.dim_sizes()[NDIMS] * DataTypeSize(dtype()));
618 dims[d] = shape_.dim_sizes()[d];
729 return shaped<T, NDIMS>(ComputeFlatInnerDims(shape_.dim_sizes(), NDIMS))
[all...] |
| tensor_shape_test.cc | 325 : shape_(shape), d_(d) {}
327 DCHECK(shape_ == rhs.shape_);
331 DCHECK(shape_ == rhs.shape_);
336 return TensorShapeDimOld(shape_->dim_size(d_));
340 const TensorShapeOld* shape_; member in class:tensorflow::__anon39597::TensorShapeIterOld
|
| tensor_shape.h | 323 TensorShapeIter(const Shape* shape, int d) : shape_(shape), d_(d) {}
325 DCHECK(shape_ == rhs.shape_);
329 DCHECK(shape_ == rhs.shape_);
333 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); }
336 const Shape* shape_; member in class:tensorflow::TensorShapeIter
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor.h | 59 shape_(shape.begin(), shape.end()),
73 : SparseTensor(other.ix_, other.vals_, other.shape_, other.order_) {}
77 std::move(other.shape_), std::move(other.order_)) {}
82 shape_ = other.shape_;
89 int dims() const { return shape_.size(); }
114 VarDimArray shape() const { return shape_; }
186 res[i] = shape_[dim_indices[i]];
216 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_[di]) valid = false;
221 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_[di]) valid = false
294 ShapeArray shape_; member in class:tensorflow::sparse::SparseTensor
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_nodes.cc | 119 : shape_(shape), type_(type), values_(values), empty_weight_flag_(false) {
124 : shape_(), type_(type), values_(nullptr), empty_weight_flag_(true) {}
127 : shape_(rhs.shape_),
134 for (int i = 0; i < shape_.nbDims; i++) c *= shape_.d[i];
144 return nvinfer1::Weights{trt_type, GetValues(), GetShapeSize(shape_)};
159 nvinfer1::Dims shape_; member in class:tensorflow::tensorrt::convert::__anon39442::TRT_ShapedWeights
200 return weights().shape_;
290 int r = iweights.shape_.d[0]
[all...] |
