| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| fake_param_op.cc | 35 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype, tensor_shape, &shape_));
40 ctx->SetOutput(0, xla::Zeros(b, shape_));
44 xla::Shape shape_; member in class:tensorflow::XlaFakeParamOp
|
| sendrecv_ops.cc | 60 xla::Shape shape_; member in class:tensorflow::__anon44172::RecvOp
72 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype, tensor_shape, &shape_));
79 ctx->SetOutput(0, xla::Recv(ctx->builder(), shape_, channel));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.h | 38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {}
61 const Shape& shape() const { return shape_; }
80 string ToString() const { return ShapeUtil::HumanStringWithLayout(shape_); }
87 Shape shape_; member in class:xla::ShapeLayout
|
| shape_tree.h | 123 const Shape& shape() const { return *shape_; }
134 shape_ = shape;
327 // If we own our Shape, this field contains it, and shape_ is a pointer into
333 const Shape* shape_; member in class:xla::ShapeTree
483 shape_(shape_storage_.get()) {
484 const int64 count = CountSubshapes(*shape_);
490 InitChildren(*shape_, &nodes_[0], &index_table_[0]);
494 ShapeTree<T>::ShapeTree(const Shape* shape) : shape_(shape) {
495 const int64 count = CountSubshapes(*shape_);
501 InitChildren(*shape_, &nodes_[0], &index_table_[0])
[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::__anon44368::TestInfeedBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| loop_emitter.h | 82 Shape shape_; member in class:xla::llvm_ir::LoopEmitter
|
| ir_array.h | 219 const Shape& GetShape() const { return shape_; }
316 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 | 38 TensorShape shape_; member in class:tensorflow::VariableOp
|
| conditional_accumulator_base_op.h | 52 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
84 PartialTensorShape shape_; member in class:tensorflow::ConditionalAccumulatorBaseOp
|
| scoped_allocator_ops.cc | 95 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
114 OP_REQUIRES(context, backing_tensor.NumElements() >= shape_.num_elements(),
118 shape_.num_elements()));
121 CHECK(output.CopyFrom(backing_tensor, shape_));
158 TensorShape shape_; member in class:tensorflow::ScopedAllocatorConcatOp
|
| scoped_allocator_ops_test.cc | 139 shape_ = shape;
157 shape_ = shape;
188 CHECK_EQ(shape_, output.shape());
199 TensorShape shape_; member in class:tensorflow::ScopedAllocatorConcatOpTest
|
| 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_);
85 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_));
100 s = context->allocate_temp(dtype_, shape_, &tmp_var->val);
123 TensorShape shape_; member in class:tensorflow::TemporaryVariableOp
|
| /external/libtextclassifier/utils/ |
| 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(); }
71 const std::vector<int> shape_; member in class:libtextclassifier3::TensorView
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xrt.h | 48 const Shape shape_; member in class:xla::swig::XrtAllocation
|
| /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/contrib/framework/kernels/ |
| zero_initializer_op.cc | 94 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &shape_));
109 dtype_, shape_, &unused, &var_tensor, attr));
135 TensorShape shape_; member in class:tensorflow::ZeroVarInitializer
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| defuser_test.cc | 42 const Shape shape_ = ShapeUtil::MakeShape(F32, {2, 2}); member in class:xla::__anon44370::DefuserTest
49 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
51 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
53 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
65 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
67 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
69 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
89 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
91 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
93 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1))
[all...] |
| hlo_matchers.h | 109 explicit HloShapeMatcher(const Shape& shape) : shape_(shape) {}
116 Shape shape_; member in class:xla::testing::HloShapeMatcher
122 explicit HloShapeAndLayoutMatcher(const Shape& shape) : shape_(shape) {}
129 Shape shape_; member in class:xla::testing::HloShapeAndLayoutMatcher
|
| indexed_array_analysis.h | 114 const Shape& shape() const override { return shape_; }
118 : operand_(operand), shape_(shape) {}
121 const Shape shape_; member in class:xla::IndexedArrayAnalysis::ReshapedArray
178 const Shape& shape() const override { return shape_; }
200 shape_(std::move(shape)) {}
206 Shape shape_; member in class:xla::IndexedArrayAnalysis::ScalarIndexedArray
|
| /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/compiler/tf2xla/ |
| xla_resource.h | 81 const TensorShape& shape() const { return shape_; }
170 TensorShape shape_; member in class:tensorflow::XlaResource
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_state_ops.h | 236 shape_ = xla::ShapeUtil::MakeTupleShape(xla_shapes);
238 shape_.Swap(&xla_shapes.front());
242 OP_REQUIRES_OK(ctx, GetShapeWithLayout(shape_, minor_to_major,
245 shape_.Swap(&shape_with_layouts);
284 shape_.IsTuple() ? xla::BorrowingLiteral(tensors_data, shape_)
285 : xla::BorrowingLiteral(tensors_data.front(), shape_);
311 xla::Shape shape_; member in class:tensorflow::XRTAllocateFromTensorOp
|
