Home
last modified time | relevance | path

Searched defs:dims_ (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dfill_test.cc70 int dims_; member in tflite::__anona9db85f40111::FillOpModel
105 int dims_; member in tflite::__anona9db85f40111::QuantizedFillOpModel
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.h93 uint16 dims_[6]; member
96 uint32 dims_[3]; member
99 gtl::InlinedVector<int64_t, 4>* dims_; member
Dshape_inference.h161 const std::vector<DimensionHandle> dims_; variable
Dtensor_shape.cc908 const int dims_ = dims(); in MergeWith() local
/external/tensorflow/tensorflow/core/util/sparse/
Ddim_comparator.h91 const int dims_; variable
Dgroup_iterator.h134 const int dims_; variable
Dsparse_tensor.h266 int dims_; variable
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.h214 std::vector<int64_t> dims_; variable
/external/tensorflow/tensorflow/core/kernels/
Dsparse_slice_op_gpu.cu.cc80 int dims_; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc101 nvinfer1::Dims dims_; member in tensorflow::tensorrt::ScopedShapedWeights
/external/tensorflow/tensorflow/lite/kernels/internal/
Druntime_shape.h185 int32_t dims_[kMaxSmallSize]; member
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc695 Dimensions dims_; member in xla::cpu::__anon8d179a540111::TiledSmallGemmEmitter::Config
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2133 DisjointSet<shape_inference::DimensionHandle> dims_; member in tensorflow::grappler::SymbolicShapeManager