Searched defs:is_batch (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding.cc | 62 bool is_batch = absl::c_linear_search( in CanFoldOperandsIntoDot() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2280 bool is_batch, int num_examples, in ParseContextDenseFeatures()
2343 bool is_batch, int num_examples, in ParseContextSparseFeatures()
2404 bool is_batch, int num_examples, in ParseContextRaggedFeatures()
2482 bool is_batch, int num_examples, in ParseSequenceDenseFeatures()
2635 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceSparseFeatures()
2763 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceRaggedFeatures()
2914 bool is_batch) { in FastParseSequenceExample()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 1080 Status MatMulGradHelper(const Scope& scope, const bool is_batch, in MatMulGradHelper()
1107 const bool is_batch, in MatMulGradCommon()
|
| D | math_grad_test.cc | 597 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { in TestMatMulGrad()
622 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, in RandMatMulShapes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 466 bool is_batch = TensorShapeUtils::IsVector(serialized->shape()); in Compute() local
|
| /external/llvm-project/llvm/utils/benchmark/include/benchmark/ |
| D | benchmark.h | 659 bool State::KeepRunningInternal(size_t n, bool is_batch) { in KeepRunningInternal()
|
| /external/libcxx/utils/google-benchmark/include/benchmark/ |
| D | benchmark.h | 694 bool is_batch) { in KeepRunningInternal()
|
| /external/llvm-project/libcxx/utils/google-benchmark/include/benchmark/ |
| D | benchmark.h | 694 bool is_batch) { in KeepRunningInternal()
|
| /external/google-benchmark/include/benchmark/ |
| D | benchmark.h | 701 bool is_batch) { in KeepRunningInternal()
|