Searched refs:SparseTensorsMap (Results 1 – 4 of 4) sorted by relevance
25 `SparseTensorsMap`.31 The container name for the `SparseTensorsMap` created by this op.37 The shared name for the `SparseTensorsMap` created by this op.41 summary: "Add a `SparseTensor` to a `SparseTensorsMap` return its handle."47 object (a `SparseTensorsMap`). A unique key within this container is generated52 the correct `SparseTensorsMap` is accessed, ensure that the same
27 `SparseTensorsMap`. Shape: `[N]`.33 The container name for the `SparseTensorsMap` created by this op.39 The shared name for the `SparseTensorsMap` created by this op.43 summary: "Add an `N`-minibatch `SparseTensor` to a `SparseTensorsMap`, return `N` handles."62 the correct `SparseTensorsMap` is accessed, ensure that the same
32 `SparseTensorsMap`.38 The container name for the `SparseTensorsMap` read by this op.44 The shared name for the `SparseTensorsMap` read by this op.46 of the Op that created the original `SparseTensorsMap` should be used.49 summary: "Read `SparseTensors` from a `SparseTensorsMap` and concatenate them."
42 class SparseTensorsMap : public ResourceBase { class44 explicit SparseTensorsMap(const string& name) : name_(name), counter_(0) {} in SparseTensorsMap() function in tensorflow::SparseTensorsMap107 ~SparseTensorsMap() override {} in ~SparseTensorsMap()120 typedef std::function<Status(SparseTensorsMap**)> CreatorCallback;131 SparseTensorsMap** sparse_tensors_map) { in GetMap()142 CreatorCallback sparse_tensors_map_creator = [this](SparseTensorsMap** c) { in GetMap()143 SparseTensorsMap* map = new SparseTensorsMap(cinfo_.name()); in GetMap()149 cinfo_.resource_manager()->LookupOrCreate<SparseTensorsMap>( in GetMap()161 SparseTensorsMap* sparse_tensors_map_ TF_PT_GUARDED_BY(mu_);173 SparseTensorsMap* map; in Compute()[all …]