Home
last modified time | relevance | path

Searched refs:IsStateful (Results 1 – 15 of 15) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFunctors.h99 static const bool IsStateful = false;
144 static const bool IsStateful = true;
224 static const bool IsStateful = false;
265 static const bool IsStateful = false;
306 static const bool IsStateful = false;
351 static const bool IsStateful = false;
375 static const bool IsStateful = false;
401 static const bool IsStateful = false;
426 static const bool IsStateful = false;
DTensorReductionCuda.h348 static const bool HasOptimizedImplementation = !Op::IsStateful &&
353 static const bool HasOptimizedImplementation = !Op::IsStateful &&
638 static const bool HasOptimizedImplementation = !Op::IsStateful &&
643 static const bool HasOptimizedImplementation = !Op::IsStateful &&
700 static const bool HasOptimizedImplementation = !Op::IsStateful &&
DTensorReduction.h245 static const bool HasOptimizedImplementation = !Op::IsStateful;
/external/tensorflow/tensorflow/core/framework/
Dop_segment.cc106 return lib->IsStateful(node_op) && in ShouldOwnKernel()
Dfunction.h847 virtual bool IsStateful(const std::string& function_name) const = 0;
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h227 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry);
228 bool IsStateful(const NodeDef node); // use OpRegistry::Global()
Dop_types.cc693 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry) { in IsStateful() function
705 bool IsStateful(const NodeDef node) { in IsStateful() function
706 return IsStateful(node, OpRegistry::Global()); in IsStateful()
Dgrappler_item.cc154 (IsStateful(node, &*fn_library) || IsDataset(node))) { in NodesToPreserve()
/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h284 static constexpr bool IsStateful = true; member
382 IsStateful = true,
392 IsStateful = true,
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc77 TEST(CAPI_EXPERIMENTAL, IsStateful) { in TEST() argument
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc178 bool IsStateful(const string& function_name) const override;
269 bool FunctionLibraryRuntimeOverlay::IsStateful( in IsStateful() function in tensorflow::FunctionLibraryRuntimeOverlay
356 bool IsStateful(const string& function) const override;
1304 bool FunctionLibraryRuntimeImpl::IsStateful(const string& func) const { in IsStateful() function in tensorflow::FunctionLibraryRuntimeImpl
Dfunction_test.cc377 TEST_F(FunctionLibraryRuntimeTest, IsStateful) { in TEST_F() argument
379 EXPECT_TRUE(flr0_->IsStateful("Variable")); in TEST_F()
380 EXPECT_TRUE(flr0_->IsStateful("VariableV2")); in TEST_F()
381 EXPECT_FALSE(flr0_->IsStateful("Matmul")); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc185 if (ctx->function_library()->IsStateful(func.name())) { in CreateShortCircuitInfo()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc891 !IsStateful(node) && !HasInputOrOutputRefs(node); in CanForceFP16()
/external/tensorflow/
DRELEASE.md1040 * The deprecated `DatasetBase::IsStateful` method is removed in favor of