Searched refs:DimensionOrConstant (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | shape_inference.h | 37 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 …]
|
D | common_shape_fns.h | 155 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,
|
D | shape_inference.cc | 677 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 …]
|
D | common_shape_fns.cc | 150 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()
|
D | shape_inference_test.cc | 123 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
|
/external/tensorflow/tensorflow/core/ops/ |
D | batch_ops.cc | 69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon8ccc74200102() 70 shape_inference::DimensionOrConstant(3)})})); in __anon8ccc74200102()
|
D | parsing_ops.cc | 49 shape_inference::DimensionOrConstant rank(c->UnknownDim()); in AddSparseOutputShapes()
|
D | array_ops.cc | 302 passed_shape, 0, c->MakeDim(shape_inference::DimensionOrConstant(1)), in __anonf6523ebd0202()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
D | xla_ops.cc | 200 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()
|