Home
last modified time | relevance | path

Searched refs:DimensionOrConstant (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h37 struct DimensionOrConstant;
68 friend struct DimensionOrConstant;
117 struct DimensionOrConstant { struct
120 DimensionOrConstant(DimensionHandle dim);
123 DimensionOrConstant(int64 val);
130 DimensionOrConstant(); argument
329 static inline int64 Value(DimensionOrConstant d) { in Value()
332 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown()
424 ShapeHandle MakeShape(std::initializer_list<DimensionOrConstant> dims);
436 ShapeHandle Vector(DimensionOrConstant dim);
[all …]
Dcommon_shape_fns.h155 DimensionOrConstant filter_size,
165 DimensionOrConstant filter_size, int64 dilation_rate, int64 stride,
221 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N,
222 const std::vector<DimensionOrConstant>& spatial,
223 DimensionOrConstant C, ShapeHandle* out,
Dshape_inference.cc677 std::initializer_list<DimensionOrConstant> dims) { in MakeShape()
680 for (const DimensionOrConstant& d : dims) { in MakeShape()
706 ShapeHandle InferenceContext::Vector(DimensionOrConstant dim) { in Vector()
710 ShapeHandle InferenceContext::Matrix(DimensionOrConstant dim1, in Matrix()
711 DimensionOrConstant dim2) { in Matrix()
953 DimensionOrConstant divisor, in Divide()
977 Status InferenceContext::Add(DimensionHandle first, DimensionOrConstant second, in Add()
1004 DimensionOrConstant second, in Subtract()
1027 DimensionOrConstant second, in Multiply()
1055 Status InferenceContext::Min(DimensionHandle first, DimensionOrConstant second, in Min()
[all …]
Dcommon_shape_fns.cc150 shape_inference::DimensionOrConstant filter_size, int64 dilation_rate, in GetWindowedOutputSizeFromDimsV2()
197 shape_inference::DimensionOrConstant filter_size, int64 stride, in GetWindowedOutputSizeFromDims()
583 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N, in MakeShapeFromFormat()
584 const std::vector<DimensionOrConstant>& spatial, in MakeShapeFromFormat()
585 DimensionOrConstant C, ShapeHandle* out, in MakeShapeFromFormat()
Dshape_inference_test.cc123 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/
Dbatch_ops.cc69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon8ccc74200102()
70 shape_inference::DimensionOrConstant(3)})})); in __anon8ccc74200102()
Dparsing_ops.cc49 shape_inference::DimensionOrConstant rank(c->UnknownDim()); in AddSparseOutputShapes()
Darray_ops.cc302 passed_shape, 0, c->MakeDim(shape_inference::DimensionOrConstant(1)), in __anonf6523ebd0202()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc200 shape_inference::DimensionOrConstant in __anon00ae58f50202()
203 shape_inference::DimensionOrConstant in __anon00ae58f50202()
234 shape_inference::DimensionOrConstant lhs_batch_dimension_or_constant( in __anon00ae58f50202()
236 shape_inference::DimensionOrConstant rhs_batch_dimension_or_constant( in __anon00ae58f50202()